PyTorch -当通过ImageFolder加载数据时，如何使用k-折交叉验证？

Question

我的数据是图像，存储在文件系统中，并通过PyTorch的ImageFolder数据加载器输入到我的卷积神经网络中。因此，training、validation和test数据被手动拆分到文件系统上的不同文件夹中。那么，在使用ImageFolder时如何应用k-fold cross validation呢

Answer 1

您可以使用data.ConcatDataset将当前拥有的固定训练/val/测试折叠合并为一个数据集。然后，您可以使用data.Subset一遍又一遍地将单个数据集随机拆分为不同的折叠。