junxnone / aiwiki

AI Wiki

https://junxnone.github.io/aiwiki

18 stars 2 forks source link

Tools VIS netron #135

Open junxnone opened 5 years ago

junxnone commented 5 years ago

模型结构可视化工具 netron

Visualizer for deep learning and machine learning models
Support model files
- ONNX (.onnx, .pb, .pbtxt)
- Keras (.h5, .keras)
- CoreML (.mlmodel)
- Caffe (.caffemodel, .prototxt)
- Caffe2 (predict_net.pb, predict_net.pbtxt)
- MXNet (.model, -symbol.json)
- PyTorch (.pth)
- Torch (.t7)
- CNTK (.model, .cntk)
- PaddlePaddle (model)
- scikit-learn (.pkl)
- TensorFlow Lite (.tflite)
- TensorFlow.js (model.json, .pb)
- TensorFlow (.pb, .meta, .pbtxt).

Install

pip install netron

Usecase

netron -b xxx.pb

http://localhost:8080/

netron xxx.onnx --port xxx_port --host xxx_host

access from http://xxx_host:xxx_port

Reference

junxnone commented 5 years ago

junxnone/tech-io#287 junxnone/ml#24