本文介绍了如何在PyCharm环境下，利用YOLOv5的export.py脚本将模型转换为ONNX格式，并通过Netron工具进行模型可视化。首先，在命令行中安装onnx包，然后运行export.py生成yolov5s.onnx文件，最后下载并使用Netron打开该文件进行可视化。
本文基于pycharm编译器下使用YOLOV5 6.0版本
1.在控制台安装onnx包
pip install onnx
2.运行export.py文件（未改动则直接在yolov5项目文件下） 默认是生成yolov5s.pt的onnx文件
3.下载Netron 直接用.exe安装到本地
https://github.com/lutzroeder/netron/releases/tag/v5.6.5
4.用Netron打开之前生成的yolov5s.onnx文件即可生成
本文参考这篇文章 学习记录https://www.pythonheidong.com/blog/article/970528/a1b0274148d63bd185a7/
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
