Netron展示pytorch模型结构

第一步：需要将pytorch模型转化为.ONNX格式，转化程序如下：

model_path = './best_model/best_model.pth'
Model = YL_Effientnat(7 + 1, False)
Model.load_state_dict(torch.load(model_path,map_location=lambda storage,loc:storage))
Model.train(False)
input= torch.randn((1,3,80,80))
torch.onnx.export(Model,input,"model.onnx",verbose=False)

第二步：安装Netron

1、通过pip install netron，安装netron库；然后，通过netron库加载pytorch模型的.onnx文件进行模型网络结构展示，代码如下：

import netron
modelpath = 'model.onnx'
netron.start(modelpath)

2、下载netron.exe可执行程序，进行安装（可执行程序可在我上传的资源中找的）。

然后，打开.onnx模型文件，即可显示如下图。

Netron展示pytorch模型结构相关推荐

【视频课】Pytorch模型分析进阶（可视化，参数量与计算量，计算速度）
课程内容本次课程内容是Pytorch的高阶使用部分,要求大家预先要熟练掌握Pytorch.总课程大纲包括Pytorch模型结构的分析,Pytorch模型速度与计算量分析,Pytorch模型可视化3部 ...
【深度学习入门到精通系列】模型结构可视化神器Netron（连.pth都可以~！)
文章目录 1 Netron 2 说明 1 Netron 目前的Netron支持主流各种框架的模型结构可视化工作,我直接给出gayhub链接: https://github.com/lutzroeder ...
PyTorch 打印模型结构、输出维度和参数信息（torchsummary）
使用 PyTorch 深度学习搭建模型后,如果想查看模型结构,可以直接使用 print(model) 函数打印.但该输出结果不是特别直观,查阅发现有个能输出类似 keras 风格 model.summ ...
Pytorch实现的LSTM模型结构
LSTM模型结构 1.LSTM模型结构 2.LSTM网络 3.LSTM的输入结构 4.Pytorch中的LSTM 4.1.pytorch中定义的LSTM模型 4.2.喂给LSTM的数据格式 4.3.L ...
【Pytorch深度学习50篇】·······第五篇：【YOLO】【1】----- YOLO V3 V4 V5的模型结构
第四篇,我们已经简单的过了一个segmentation网络,下面我们要进入一个相对要复杂一点的系列了,它也是用于目标检测的,与segmentation不同的是,这个网络会回归出目标的位置和大小.YOL ...
Intel发布神经网络压缩库Distiller：快速利用前沿算法压缩PyTorch模型
Intel发布神经网络压缩库Distiller:快速利用前沿算法压缩PyTorch模型原文:https://blog.csdn.net/u011808673/article/details/8079 ...
PyTorch模型部署：pth转onnx跨框架部署详解+代码
文章目录引言基础概念 onnx:跨框架的模型表达标准 onnxruntime:部署模型的推理引擎示例代码 0)安装onnx和onnxruntime 1)pytorch模型转onnx模型 2)on ...
PyTorch 模型训练实用教程(附代码)
向AI转型的程序员都关注了这个号???????????? 机器学习AI算法工程公众号:datayx PyTorch 能在短时间内被众多研究人员和工程师接受并推崇是因为其有着诸多优点,如采用 Py ...

Netron展示pytorch模型结构

Netron展示pytorch模型结构相关推荐

最新文章

热门文章