Netron是一个模型的展示工具,它有网页版和app版:
网页版:Netron
app版:GitHub - lutzroeder/netron: Visualizer for neural network, deep learning, and machine learning models
直接用网页版吧,还不用安装。
它可以打开不同的网络模型,比如yolov5s.pt:
然后他就会跟你生成一个图:
以上是直接用的“.pt”文件转换。
接下来看看.torchscript:(.pt效果不如.torchscript,上图就是.pt,会发现他的生成图很简陋)
yolov5源码里边的export.py文件就是转换脚本:
终端直接运行:
python export.py --weights yolov5s.pt --img 640 --batch 1
完了之后在项目的根目录就会多出这样一个文件:
ok,然后去Netron网站上进行转换:(注意左下角勾选全部文件)
点击打开之后:
可以看到生成的图比.pt丰满多了。
然后在网站的左上角可以进行展示调节:(比如展示属性,权重,横着看等等)
ok。
其实export脚本还可以转换为.onnx格式,如果要在不同的框架中使用模型进行推理,建议选择.onnx格式的(需要pip install onnx一下);如果只是在pytorch中使用,则可以选择.torchscript格式的。