python 文件夹复制

小工具，文件夹内容过大，只选取前N张图片。

import os
from shutil import copy
i = 0
save_dir = 'E:/test'
if not os.path.isdir(save_dir):
    os.mkdir(save_dir)
file_path = 'E:\dataset\VOC07+12\JPEGImages'
img_name = os.listdir(file_path)
img_name.sort()
for img in img_name:
    img_path = os.path.join(file_path, img)
    copy(img_path, save_dir)
    i += 1
    if i > 9:
        break