如何使用PyTorch从一个目录将数据分为训练集和测试集？

Question

我有一个数据文件夹，其中没有将数据拆分为训练和测试文件夹。如何将数据分为训练集和测试集？标签来自文件名，因此按此顺序进行的任何更改都必须包含标签。我想在使用ImageFolder之前拆分数据，以便可以对训练和测试数据集进行不同的转换。

reposurgeon

Answer 1

我认为您正在寻找的是交叉验证，请选中此answer。您可以添加带有标签的列，然后应用任何交叉验证方法将其分为测试和训练。