Мне интересно, какой самый простой способ заставить модель UNet работать с переменным входным разрешением? Я имею в виду, что если я уменьшу выборку 3 раза, то, например, для изображения размером 320x320 пространственные размеры векторных слоев будут:
320 -> 160 -> 80 -> 40 -> 80 -> 160 -> 320
С другой стороны, если я введу изображение размером, скажем, , 960x540, тогда я получу
540 -> 270 -> 135(!) -> 68 -> 136(!) ->...
Сеть взорвется с таким входным изображением, потому что размеры тензора имеют несоответствие при конкатенации тензора из пропуска.
Есть ли некоторые стандартная уловка для обработки этого?