import os
import shutil

# 指定包含标注txt文件的文件夹和目标文件夹
txt_folder = 'C:/Users/90458/Downloads/EgoDexter/processed_data_4/'  # 包含标注txt文件的文件夹
image_folder = 'C:/Users/90458/Downloads/EgoDexter/data/Rotunda/color/'      # 图像文件所在的文件夹
output_folder = 'C:/Users/90458/Downloads/EgoDexter/processed_data_4_img/'    # 目标文件夹，用于存放找到的图像文件

# 确保目标文件夹存在，如果不存在则创建
if not os.path.exists(output_folder):
    os.makedirs(output_folder)

# 遍历包含标注txt文件的文件夹
for root, _, filenames in os.walk(txt_folder):
    for filename in filenames:
        if filename.endswith('.txt'):
            # 构造对应的图像文件名
            image_filename = os.path.splitext(filename)[0] + '.png'

            image_path = os.path.join(image_folder, image_filename)

            # 检查图像文件是否存在
            if os.path.exists(image_path):
                # 复制图像文件到目标文件夹
                shutil.copy(image_path, os.path.join(output_folder, image_filename))
            else:
                print(f"找不到对应的图像文件：{image_filename}")

print("完成复制操作。")