Я работаю над проектом, использующим модель глубокого обучения keras, которую мне нужно перенести в PyTorch. Целью проекта является локализация некоторых элементов на изображениях. Чтобы обучить его, я сначала использую патчи, извлеченные из моих изображений, а затем делаю вывод о полном изображении. Я прочитал, что это было возможно с (None, None, 1) входной формой для входного слоя keras, и это в настоящее время работает. Тем не менее, та же система обучения, кажется, не работает в Pytorch. Поэтому мне было интересно, что входной слой (Нет, Нет, 1) делает что-то конкретное, когда я начинаю выводить полные изображения?
Спасибо за ваши ответы