利用python划分训练集和测试集方法如下
from sklearn.model_selection import train_test_split
from PIL import Image
import os
path="./path/your_datasets" #你要划分的数据集
dat= os.listdir(path)
train, test = train_test_split(dat, test_size = 0.2)#0.2表示需要划分的测试集比例
for file in train:
im2 = Image.open("./path/your_datasets"+file)
im2.save("./path/train/"+file)#划分后的训练集存放位置
for file in test:
im2 = Image.open("./path/your_datasets"+file)
im2.save("./path/test/"+file)#划分后的训练集存放位置
根据注释改成自己的路径,运行结束就可以得到自己随机划分的训练集和测试集了。