| 1234567891011121314151617181920212223242526272829303132333435 |
- import os
- import shutil
- # 输入文件夹路径:只需修改这三行
- json_folder = r"G:\python_ws_g\data\test\jsonjpg"
- tiff_folder = r"G:\python_ws_g\data\test\tiff"
- output_folder = r"G:\python_ws_g\data\test\tiff_json"
- # 创建输出文件夹(如不存在)
- os.makedirs(output_folder, exist_ok=True)
- # 提取 json 文件中的标识前缀(用于匹配)
- json_files = [f for f in os.listdir(json_folder) if f.endswith('_color.json')]
- json_keys = {f.replace('_color.json', ''): f for f in json_files}
- # 遍历 tiff 文件,尝试匹配 json
- matched_count = 0
- for tiff_name in os.listdir(tiff_folder):
- if not tiff_name.endswith('.tiff'):
- continue
- if '_rgbd.tiff' not in tiff_name:
- continue
- key = tiff_name.replace('_rgbd.tiff', '')
- if key in json_keys:
- # 匹配成功,复制两份文件到目标目录
- json_src = os.path.join(json_folder, json_keys[key])
- tiff_src = os.path.join(tiff_folder, tiff_name)
- shutil.copy(json_src, os.path.join(output_folder, json_keys[key]))
- shutil.copy(tiff_src, os.path.join(output_folder, tiff_name))
- print(f"匹配并复制:{json_keys[key]} <-> {tiff_name}")
- matched_count += 1
- print(f"\n✅ 共复制 {matched_count} 对文件到:{output_folder}")
|