Netron【.pt转.torchscript模型展示】

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格式的。