У меня есть набор данных изображений для классификации. в каталоге набора данных у меня есть два подкаталога со всеми поездами и тестовыми изображениями и два файла csv с именем изображения и класса (только для данных о поездах, но не для тестовых данных).
Пример: в каталоге набора данных - -> / content / dataset / Train Images / image10.jpg путь для одиночного изображения каталога изображений поездов. Файл csv содержит: Класс изображения 0 image7042.jpg Пища
теперь позволяет получить сопоставление между этим файлом csv и то же самое имя изображения, что и "Train Images". это необходимо для создания набора данных с тензорным потоком и последующего обучения модели cnn. Решение Pytorch также будет оценено. заранее спасибо.