今天发现了一个模型可视化的神器-netron,
只需下载安装,即可用,目前支持supports ONNX (.onnx
, .pb
, .pbtxt
), Keras (.h5
, .keras
), CoreML(.mlmodel
), Caffe2 (predict_net.pb
, predict_net.pbtxt
), MXNet (.model
, -symbol.json
) and TensorFlow Lite (.tflite
). Netron has experimental support for Caffe(.caffemodel
, .prototxt
), PyTorch (.pth
), Torch (.t7
), CNTK (.model
, .cntk
), PaddlePaddle (__model__
), Darknet (.cfg
), scikit-learn (.pkl
), TensorFlow.js(model.json
, .pb
) and TensorFlow (.pb
, .meta
, .pbtxt
).
支持各种平台