Unet网络下采样出现奇数分辨率导致torch.cat报错

encourageykk

于 2023-11-04 12:28:47 发布

阅读量347

点赞数

文章标签： python 深度学习人工智能

本文链接：https://blog.csdn.net/encourageykk/article/details/134216727

版权

在搭建完一个5层深的Unet网络后，输入网络输入2D图像的shape = (378, 512)，代码报错，不能torch.cat()，在dim=2上95 and 96维度不同的。检查发现尺寸378下采样过程：378 --> 189 --> 95 --> 48–> 24 --> 12，其上采样过程为12 --> 24 --> 48 --> 96 --> 192，因此在跳跃连接时出错。解决方案时将378尺寸pading到最接近的偶数尺寸，即12 --> 24 --> 48 --> 96 --> 192 --> 384, 输入图像从shape = (378, 512) --> (384, 512).