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/

猜你喜欢

转载自blog.csdn.net/qq_40992227/article/details/119172266