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

最新推荐文章于 2026-03-24 00:22:49 发布

原创最新推荐文章于 2026-03-24 00:22:49 发布 · 783 阅读

4 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

收录于

当前文章被以下社区和专栏收录：

在构建深度为5的Unet网络时，遇到torch.cat错误，因输入图像尺寸(378,512)与网络结构的上、下采样步长不匹配。通过将输入图像大小调整为偶数尺寸(384,512)，解决了维度问题。

在搭建完一个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).

标签