欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

netron可视化模型

程序员文章站 2022-06-05 22:05:27
...

1. netron的github(https://github.com/lutzroeder/Netron

2. 在conda虚拟环境中安装

#指定清华源安装
pip install netron -i  https://pypi.tuna.tsinghua.edu.cn/simple 

3 新建运行py文件;

import netron
modelPath = "model.pb"
netron.start(modelPath)

 

相关标签: 脚本程序