Я хотел бы использовать ImageFolder для создания набора данных изображений.
Моя текущая структура каталогов изображений выглядит следующим образом:
/root
-- train/
---- 001.jpg
---- 002.jpg
---- ....
-- test/
---- 001.jpg
---- 002.jpg
---- ....
Я хотел бы иметь набор данных, предназначенный для тренировочных данных, и набор данных, предназначенный для тестовых данных.
Как я понимаю, делаю так:
dataset = ImageFolder(root='root/train')
не находит изображений.
Doing
dataset = ImageFolder(root='root')
найти изображения, но train и test изображения просто скремблированы.
ImageFolder имеет аргумент loader, но мне не удалось найти ни одного варианта использования для него.
Как различить изображения в корневой папке по подпапке, к которой они принадлежат?