网络可视化工具netron
netron可视化工具,可以清晰的看到每一层的输入输出,网络总体的架构,而且支持各种不同网络框架,简单好用
在线版网站 : netron
以yolov5s为例
使用.pt展示
(.pt文件展示效果不如onnx)
使用onnx展示
安装onnx
转换得到onnx文件
使用Yolov5代码中models/export.py脚本
下载win版
MAC和Linux版本
参数解读
yolov5的激活函数如下:
对应到网络中是:
- 在Conv中W(32x12x3x3),
12是图像输入的channel数(之前操作将4个3通道的拼接在一起了),
32是输出图像的channel
3x3是卷积核的大小 - 在add和div中,B都是操作数
后面的操作同理:
(注意,解析的网络结构图,可以展示当前操作的卷积参数,但不能显示图像在当前层的维度是多少。。。)