更换模型时,经常出现这个问题,大概率是通道不匹配,首先确认自己的图片是黑白还是彩色,其次看通道,首先找到根据报错断点位置,发现断点之后,加一行print(x.shape)查看当前的尺寸,按照这个尺寸进行调整即可。下面是我碰到的问题该错过程,我是在efficientnet中更换模型b2时产生错误,但是参数都来自官网,我不知道为什么报错,不过可以解决这个问题。
具体参数可以看这位兄弟的博客(17条消息) EfficientNet B0-B7 网络参数_繁华落尽,寻一世真情的博客-CSDN博客_efficientnet参数量下面是我的详细改错过程
报错在这里
找到这里
打印出来的尺寸,竟然是这样的,按照这个更改
改成这样
如此图所示,程序开始正常运行了