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导入即可查看模型结构图

GitHub 加速计划 / ne / netron
27.07 K
2.72 K
下载
lutzroeder/netron: 是一个用于查看和可视化神经网络模型的桌面应用程序,支持多种深度学习框架和常用的神经网络格式。适合用于可视化神经网络模型,尤其是对于需要进行神经网络模型调试和可视化的开发人员和研究人员。
最近提交(Master分支:1 个月前 )
9cee7758 - 3 个月前
63a272e8 - 3 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐