你是不是在Pycharm等编译器和Colab等服务器上运行yolov5的detect.py出现下面的这个问题
那么我们跟着错误去上面的提示信息找方法
可以发现是关于”Upsample“和 ‘recompute_scale_factor’,相关的提示是在这里
紧接着 ,按着提示文件路径一步一步的打开usampling.py,找到第155行附近
这是发现在,154行有我们刚才提到的关键信息
然后把第154行还有153行最后的逗号“,”删掉,再保存一下子
最后的效果如下
或者你把刚才提到的哪一行加个注释,注释掉也是可以的。
def forward(self, input: Tensor) -> Tensor:return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)
然后运行下detect.py,发现问题解决~
更多干货,可以关注公众号“蓝胖胖干货”,教你账号注册,快速上网等秘籍~~~咱们一起成长!