本文分享使用pytorch可视化工具netron查看YOLOv5网络模型结构。
正常在控制台pip命令安装
pip install netron
执行netron
C:\Users>netron
Serving at http://localhost:8080
网页打开
在YOLOv5中，打开网络模型主文件yolo.py。
将模型保存为一个pt文件
# Create model
model = ModelYOLOv5 完整网络模型可视化 pytorch netron
最新推荐文章于 2025-10-24 16:52:56 发布
本文分享使用pytorch可视化工具netron查看YOLOv5网络模型结构的方法。先通过pip命令安装并执行netron，在网页打开。在YOLOv5中保存模型为pt文件，拖入网页查看。若想看到完整结构，可创建输入x，用torch.jit.trace()跟踪网络。
最低0.47元/天 解锁文章
扫一扫
1639
到【灌水乐园】发言
