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