使用 Netron 可以直接查看神经网络,这里以 onnx 网络为例。

本文介绍,在 ubuntu 下安装,使用浏览器打开 onnx 网络。

Netron地址

https://github.com/lutzroeder/Netron

安装

python -m pip install netron

打开onnx

# create script: test.py
import netron 

model_path = "new_resnet50_bw.onnx"
netron.start(model_path)

# run
python test.py

浏览器打开

# 地址栏输入
http://localhost:8080/

网络局部图

在这里插入图片描述

参考

安装方式:https://github.com/lutzroeder/netron/issues/438

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐