YOLOv5导出onnx遇到的坑

1、运行export.py时报错 ‘Upsample’ object has no attribute ‘recompute_scale_factor’
解决办法：找到Upsampling.py文件（我的文件在这个路径下找到的，仅供参考D:\Program Files\Lib\site-packages\torch\nn\modules），将forward函数修改如下：

def forward(self, input: Tensor) -> Tensor:
        return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners) #把括号中的这一部分删掉即可,recompute_scale_factor=self.recompute_scale_factor

2、以上修改完成后运行export.py，程序不再报错，但未能导出onnx，提升未安装onnx，并提示使用如下命令进行安装：

pip install 'onnx'

按照提升的命令安装时，速度很慢，采用清华镜像则可以秒装，命令如下：

pip install 'onnx'  -i https://pypi.tuna.tsinghua.edu.cn/simple/

安装好onnx后，再次运行export.py，成功导出onnx文件。
导出onnx文件后，按照程序的提示，使用这个链接 https://netron.app可以打开onnx文件，我打开后如下图所示：