你是不是在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,发现问题解决~

更多干货,可以关注公众号“蓝胖胖干货”,教你账号注册,快速上网等秘籍~~~咱们一起成长!