根据txt图片名将文件夹的图片复制到其他文件夹

# -*- coding: utf-8 -*-

import shutil
#这个库复制文件比较省事
  
def objFileName():
 '''
 生成文件名列表
 :return:
 '''
 local_file_name_list = r'./ImageSets/Segmentation/trainval.txt'
 #指定名单
 obj_name_list = []
 for i in open(local_file_name_list,'r'):
  obj_name_list.append(i.replace('\n',''))
 return obj_name_list
  
def copy_img():
 '''
 复制、重命名、粘贴文件
 :return:
 '''
 local_img_name=r'./JPEGImages'
 #指定要复制的图片路径
 path = r'./zongtu'
 #指定存放图片的目录
 for i in objFileName():
  new_obj_name = i+'.jpg'
  shutil.copy(local_img_name+'/'+new_obj_name,path+'/'+new_obj_name)
  
if __name__ == '__main__':
 copy_img()