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("完成复制操作。")
python寻找txt数据集标签文件中对应的图片并输出到另一个文件夹里面
于 2023-09-08 20:26:44 首次发布