MMdnn Visualizer網絡可視化。

微軟最近開源了MMdnn，可用于轉換、可視化和診斷深度神經網絡。目前支持 CNTK、Tensorflow、Caffe、Keras、MXNet等框架。

https://github.com/Microsoft/MMdnn

?

具體步驟：

1 安裝MMdnn

pip install https://github.com/Microsoft/MMdnn/releases/download/0.1.3/mmdnn-0.1.3-py2.py3-none-any.whl

相關依賴

pip install tensorflow pip install keras conda install -c anaconda requests

2 用keras編譯模型

以keras自帶的example，mnist_mlp.py為例

model = Sequential() model.add(Dense(512, activation='relu', input_shape=(784,))) model.add(Dropout(0.2)) model.add(Dense(512, activation='relu')) model.add(Dropout(0.2)) model.add(Dense(num_classes, activation='softmax')) model.summary() model.compile(loss='categorical_crossentropy', optimizer=RMSprop(), metrics=['accuracy'])

3 將模型保存為.json格式

json_string = model.to_json() with open('mlp_model.json','w') as of:of.write(json_string)

4 將模型轉換為中間表示

在mlp_model.json文件目錄下conda terminal執行

python -m mmdnn.conversion._script.convertToIR -f keras -d keras_mlp -n mlp_model.json

5 瀏覽器打開MMdnn Visualizer并選擇文件

http://mmdnn.eastasia.cloudapp.azure.com:8080/

暫時不清楚是否可以保存

總結

以上是生活随笔為你收集整理的MMdnn Visualizer网络可视化的全部內容，希望文章能夠幫你解決所遇到的問題。

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