該代碼可以將file_path_src文件夾中的前cnt個文件,剪切或復制到file_path_tar文件夾下,前提是file_path_src中的文件名可以排序。如VOC數據集提取某個類的圖片和xml后,將他們划分並放在train和val文件夾下時,可以用到以下程序。
import os import shutil file_path_src = r'/home/×××/×××/×××' #源目錄 file_path_tar = r'/home/×××/×××/×××' #目標目錄 cnt = 0 for 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