神经网络可视化工具 Netron

使用 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