你是不是在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,发现问题解决~
就是把没有的模块删掉就好了~~~