yolov5可视化算法网络整体框架
文章目录
- 前言
- 问题
- 解决方式
前言
问题
可视化算法网络整体框架：yolov5是网络文件格式，不同与yolov3、4的cfg可视化格式。因此无法用netron工具直接可视化查看网络结构.pt。如下图所示为netron直接查看.pt文件的部分截图。
解决方式
采用pt->onnx->netron折中格式。
首先使用yolov5中models->export.py脚本将pt文件转换成onnx格式，再用netron工具打开转换后的models->.onnx文件，即可看到全网络的整体架构。如下图所示整体架构的部分截图。
提示：
netron网页地址：https://netron.app/