from sklearn.model_selection import train_test_split
from PIL import Image
import os
path="C:/Users/86152/OneDrive/桌面/result"
dat= os.listdir(path)
train, test = train_test_split(dat, test_size = 0.1)
for file in train:
im2 = Image.open("C:/Users/86152/OneDrive/桌面/result/"+file)
im2.save("C:/Users/86152/OneDrive/桌面/train_result/"+file)
for file in test:
im2 = Image.open("C:/Users/86152/OneDrive/桌面/result/"+file)
im2.save("C:/Users/86152/OneDrive/桌面/test_result/"+file)
python 划分数据集(图像)使用train_test_split
最新推荐文章于 2023-06-02 19:10:33 发布