PyTorch - Как использовать перекрестную проверку в k-кратном порядке, когда данные загружаются через ImageFolder?

Question

Мои данные, которые являются изображениями, хранятся в файловой системе и передаются в мою сверточную нейронную сеть через ImageFolder загрузчик данных PyTorch. Поэтому данные training, validation и test вручную разделяются на разные папки в файловой системе. Итак, как я могу применить k-fold cross validation при использовании ImageFolder?

Answer 1

Вы можете объединить фиксированные сгибы поезда / val / test, которые у вас есть в данный момент, используя data.ConcatDataset в один набор данных. Затем вы можете использовать data.Subset, чтобы случайным образом разбить один набор данных на разные сгибы снова и снова.