def generate_path(file_path):
img_to_path = {}
for parent, dirnames, filenames in tqdm(os.walk(file_path)):
for filename in filenames:
if '.jpg' in filename:
# print(os.path.join(parent, filename))
img_to_path[filename] = os.path.join(parent, filename))
return img_to_path
def move_pic_to_images(img_to_path):
with open('./openset.txt','r') as f:
contents = f.readlines()
for content in tqdm(contents):
content_data = content.strip()
img_name = content_data.split(' ')[0]
img_path = img_to_path[img_name]
shutil.copy(img_path,'/data/benchmark/data/mall/images')
遍历文件夹中所有图片,并将符合条件的图片移动到指定位置
最新推荐文章于 2021-11-23 23:05:50 发布