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

【问题标题】：How to split folder with images into train, val and test?如何将带有图像的文件夹拆分为训练、验证和测试？
【发布时间】：2021-12-10 12:41:29
【问题描述】：

我正在使用 colab，并且我有一个包含图像的文件夹。如何将它们分成三个文件夹，其中包含随机分割的图像？我希望训练中有 0.8，val 中有 0.1，测试中有 0.1。我试过 splitfolders 库：

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

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

非常欢迎 Pytorch 解决方案

【问题讨论】：

您是否从谷歌驱动器安装了这些文件？我只是在这里用一些随机数据试了一下，它奏效了。
@DouglasFerreira no with wget from dropbox
尝试在没有内容前缀的情况下运行，并确保文件夹结构被正确标记和填充
成功了吗？有更新吗？

标签： python python-3.x directory pytorch google-colaboratory

【解决方案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))

【讨论】：