netron配置和使用

参考博文
链接: https://blog.csdn.net/qq_38253797/article/details/118415472.

一、生成文件代码

import torch
from models.yolo import Model

cfg = "./yolov5s.yaml"
# Create model
model = Model(cfg).to("cpu")
x = torch.randn(1, 3, 640, 640).to("cpu")
script_model = torch.jit.trace(model, x)
script_model.save("../weights/m_yolov5s.pt")

二、导入

进入netron官方网页连接
链接: https://netron.app/.

点击Open Model 将一中生成的m_yolov5s.pt导入即可查看模型结构图