参考:
python随机选取10000张图片并复制到另一个文件夹中 - Suii_v5的博客 - CSDN博客 https://blog.csdn.net/Suii_v5/article/details/72730792
IPython
import os
import random
import shutil
fileDir = './train/boardcast/'
tarDir = './valid/boardcast/'
rate = 0.2
pathDir = os.listdir(fileDir) #scan
filenumber = len(pathDir)
picknumber = int(filenumber * rate)
print('total {} pictures'.format(filenumber))
print('moved {} pictures to {}'.format(picknumber, tarDir))
total 1023 pictures
moved 204 pictures to ./valid/boardcast/
sample = random.sample(pathDir, picknumber)
for name in sample:
shutil.move(fileDir + name, tarDir + name)
print(name)
print('succeed moved {} pictures from {} to {}'.format(picknumber, fileDir, tarDir))
……
2018den_boardcast_7.jpg
2018fco_boardcast_22.jpg
2016rio_boardcast_25.jpg
succeed moved 204 pictures from ./train/boardcast/ to ./valid/boardcast/
完整代码
import os
import random
import shutil
fileDir = './train/court/'
tarDir = './valid/court/'
rate = 0.2
pathDir = os.listdir(fileDir) #scan
filenumber = len(pathDir)
picknumber = int(filenumber * rate)
print('total {} pictures'.format(filenumber))
print('moved {} pictures to {}'.format(picknumber, tarDir))
sample = random.sample(pathDir, picknumber)
for name in sample:
shutil.move(fileDir + name, tarDir + name)
print(name)
print('succeed moved {} pictures from {} to {}'.format(picknumber, fileDir, tarDir))