一、说明
Netron是一种用于神经网络、深度学习和机器学习模型的可视化工具,它可以为模型的架构生成具有描述性的可视化(descriptive visualization)。
支持的模型格式:
ONNX: squeezenet
TensorFlow Lite: yamnet
TensorFlow: chessbot
Keras: mobilenet
TorchScript: traced_online_pred_layer
Core ML: exermote
Darknet: yolo
Github:https://github.com/lutzroeder/netron
二、安装使用
在线版本
https://lutzroeder.github.io/netron/
操作系统
windows、mac等操作系统的可执行文件
python方式
# 安装
pip install netron
# 启动
netron
# 浏览器访问
http://localhost:8080/