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

Question

我的数据，即图像，存储在文件系统中，并ImageFolder通过PyTorch. 因此，training、validation和test数据被手动拆分到文件系统上的不同文件夹中。k-fold cross validation那么，使用时如何申请ImageFolder呢？

Answer 1

您可以将当前使用的固定训练/验证/测试折叠合并data.ConcatDataset到一个数据集中。然后，您可以使用data.Subset将单个数据集反复随机拆分为不同的折叠。