PyTorch - загрузка изображений без подпапок

Question

Прежде всего, я хотел сказать, что я новичок в PyTorch, поэтому я прошу прощения за советы, если уровень моих вопросов не так высок.Мне было интересно, если вы можете помочь мне с чем-то (на самом деле у меня есть 2 вопроса).История позади них: я работаю над классификацией изображений.Мои тестовые данные разделены на подпапки по их меткам, и я загружаю их через DataLoader.

Первый вопрос: 1) Верно ли, что если вы обучили свою модель конкретному размеру партии, протестируйте ее с другимиразмеры будут влиять на точность?2) Есть ли способ загрузить и использовать модель с тестовыми данными, расположенными в одной папке (без подпапок).Насколько мне известно, DataLoader нужны подпапки.

Заранее спасибо!

Answer 1

1. Это зависит от того, используете ли вы операции, которые зависят от других элементов в пакете.Если вы используете такие вещи, как пакетная нормализация , это возможно, но в целом, если ваша сеть обрабатывает пакетные элементы отдельно, это не так.

2. Если вы проверитедокументация torch.utils.data.Dataset , вы увидите, что для набора данных по существу требуются только методы __len__ и __getitem__, где первый говорит, сколько элементов содержит набор данных, а второй получаетi й элемент - будь то изображение и метка, изображение и его маска сегментации, или другие вещи.Ничто не мешает вам написать собственный набор данных.Я предлагаю вам взглянуть на исходный код DatasetFolder и изменить его в соответствии с вашими потребностями.