我有一个非常类似UNet的网络，但是，当我输入3x 1x 1（channel =3，height =1，width=1）的输入大小时，模型崩溃了，因为第一个最大池（kernel size =2，stride =2）会将维数减少到3x 0x 0。我如何修改Unet模型，使它可以接受我的3x 1x 1输入，并处理任意数量的池？任何帮助都是感谢！
0wi1tuuw1#
必须对预处理后的图像大小进行归一化，请参见torchvision.transforms.functional.resize。
torchvision.transforms.functional.resize
1条答案
按热度按时间0wi1tuuw1#
必须对预处理后的图像大小进行归一化，请参见
torchvision.transforms.functional.resize。