该代码可以将file_path_src文件夹中的前cnt个文件,剪切或复制到file_path_tar文件夹下,前提是file_path_src中的文件名可以排序。如VOC数据集提取某个类的图片和xml后,将他们划分并放在train和val文件夹下时,可以用到以下程序。
import osimport shutilfile_path_src = r'/home/×××/×××/×××' #源目录file_path_tar = r'/home/×××/×××/×××' #目标目录cnt = 0for files in os.walk(file_path_src): sorted_files = sorted(files[2]) for file in sorted_files: print file + "-->star" mv_files = os.path.join(file_path_src, file) shutil.move(mv_files, file_path_tar) #剪切,复制的话换成shutil.copy即可 cnt += 1 if cnt == 8000: #复制的文件个数 break