該代碼可以將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
