如下,所需要查找的文件名person_label文件夹下,读取该文件夹里的文件名称,放到names列表里。再从源数据文件夹src_img_path下查找所需要的文件(此处为图片),然后将其复制出来到另一文件夹中。
import os
import shutil
a = os.listdir("person_label")
names=[]
for i in a:
i = i.replace("txt","jpg")
names.append(i)
src_img_path='F:\datasets\coco\images/val2017'
to_img_path='F:\yolov5-master\data/img2'
for name in names:
for file in os.listdir(src_img_path):
if os.path.isfile(src_img_path+'/'+file):
if name in file:
shutil.copy(src_img_path + '/' + file, to_img_path + '/'+file)
print(name)