Leo-physique 2022-07-26 21:33 采纳率: 50%
浏览 43
已结题

pytorch复现文章的一个小报错

pytorch 文章复现的小报错
复现了一个backbone,第一次卷积就报错说我输入的图像通道有问题。具体见下图

img

说我的输入是16 * 32 * 510 * 510.

可以我的batch确实是16,但是我的图像是3 * 512 * 512,怎么可能是32 * 510 * 510呢。

于是我打印了一下transform之后的inputs的shape

img

没有发现任何问题。

报错位置代码

就是第一层卷积,扩一下通道

img

肿么了

  • 写回答

3条回答 默认 最新

  • Katsu-S 2022-07-26 22:09
    关注

    输入[16, 3, 512, 512],经过第一层卷积后变成[16, 32, 510, 510],这个错误应该说的是后面,看看你第二层卷积的输入通道是不是32

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(2条)

报告相同问题?

问题事件

  • 已结题 (查看结题原因) 7月27日
  • 已采纳回答 7月27日
  • 修改了问题 7月26日
  • 创建了问题 7月26日

悬赏问题

  • ¥60 优博讯DT50高通安卓11系统刷完机自动进去fastboot模式
  • ¥15 minist数字识别
  • ¥15 在安装gym库的pygame时遇到问题,不知道如何解决
  • ¥20 uniapp中的webview 使用的是本地的vue页面,在模拟器上显示无法打开
  • ¥15 网上下载的3DMAX模型,不显示贴图怎么办
  • ¥15 关于#stm32#的问题:寻找一块开发版,作为智能化割草机的控制模块和树莓派主板相连,要求:最低可控制 3 个电机(两个驱动电机,1 个割草电机),其次可以与树莓派主板相连电机照片如下:
  • ¥15 Mac(标签-IDE|关键词-File) idea
  • ¥15 潜在扩散模型的Unet特征提取
  • ¥15 iscsi服务无法访问,如何解决?
  • ¥15 感应式传感器制作的感应式讯响器