本文已同步至公眾號，歡迎關注。

?

很多同學在搭建深度學習模型的時候，困惑于如何查看模型結構，也困惑于如何查看模型參數。本文就對其中的部分做介紹。

?

1. 如何查看模型結構

一般采用不同的深度學習框架，都會有相應的方法可以查看對應的模型結構，但是，不同的框架，采用不同的方法，有的方法還要經過很多次配置，這就給使用造成了很多麻煩。

而且，如果手里只有別人的模型文件，模型的結構就無法得知了。

現在博主就介紹一個工具，可以通過模型文件，來查看網絡結構。如果你想仿制別人的模型，在只有模型文件的情況下，就可以輕松百飄。

上鏈接

https://github.com/lutzroeder/Netron

你只需要打開模型文件，就可以看到網絡結構。而且

Netron supports?ONNX?(.onnx,?.pb,?.pbtxt),?Keras?(.h5,?.keras),?TensorFlow Lite?(.tflite),?Caffe?(.caffemodel,?.prototxt),?Darknet?(.cfg),?Core ML?(.mlmodel),?MNN?(.mnn),?MXNet?(.model,?-symbol.json),?ncnn?(.param),?PaddlePaddle?(.zip,?__model__),?Caffe2?(predict_net.pb),?Barracuda?(.nn),?Tengine?(.tmfile),?TNN?(.tnnproto),?RKNN?(.rknn),?MindSpore Lite?(.ms),?UFF?(.uff).

Netron has experimental support for?TensorFlow?(.pb,?.meta,?.pbtxt,?.ckpt,?.index),?PyTorch?(.pt,?.pth),?TorchScript?(.pt,?.pth),?OpenVINO?(.xml),?Torch?(.t7),?Arm NN?(.armnn),?BigDL?(.bigdl,?.model),?Chainer?(.npz,?.h5),?CNTK?(.model,?.cntk),?Deeplearning4j?(.zip),?MediaPipe?(.pbtxt),?ML.NET?(.zip),?scikit-learn?(.pkl),?TensorFlow.js?(model.json,?.pb).

?

爽不爽只有用了才知道。看個圖：加載onnx模型。

?

2. 如何查看keras保存模型.h5文件內容

你肯定知道，.h5文件保存的就是模型的參數，但是里面到底是什么樣的呢？

因為.h5文件是按照層次數據格式第5代的版本(Hierarchical Data Format,HDF5)來保存模型的。

先上鏈接：

https://s3.amazonaws.com/hdf-wordpress-1/wp-content/uploads/manual/HDFView/3.1/HDFView-3.1.0-win10vs14_64.zip

我們來看個效果。

打開之后，雙擊文件名，test.h5,在右側會顯示模型文件的基本信息。其中，model_config就是模型中各個層的詳細信息。如何把它保存為txt詳細查看。

model_weights 下面有很多子選項，這些跟model_config中是一一對應的。里面保存的是參數數據。

optimizer_weights 是訓練優化數據，保存的是訓練過程中的參數，比如說adam優化器的參數。

?

?

?

?

?

總結

以上是生活随笔為你收集整理的深度学习模型如何查看（hdfview + netron）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。