python - 如何将带有图像的文件夹拆分为训练、验证和测试？

Question

我正在使用 colab，我有一个包含图像的文件夹。如何将它们分成三个带有随机分割图像的文件夹？我希望 train 中有 0.8，val 中有 0.1，test 中有 0.1。我试过 splitfolders 库:

splitfolders.ratio("content/data", output="output", seed=1337, ratio=(.8, .1, .1), group_prefix=None) 

但是没有出现新的文件夹。目前还不清楚他们的名字是什么。怎么做？还有其他解决方案吗？

Pytorch 解决方案非常受欢迎

Answer 1

在你的电脑上试试这个。

我正在我的项目中使用这段代码

import splitfolders 

#### input dataset that want to split
input_folder = 'D:/Raw_DS'  

output_folder= 'D:/Splitted_DS'

splitfolders.ratio(input_folder, output= output_folder, seed=1337, ratio = (0.8, 0.1, 0.1))