Keras:網路結構視覺化
MMdnn Visualizer網路視覺化。
微軟最近開源了MMdnn,可用於轉換、視覺化和診斷深度神經網路。目前支援 CNTK、Tensorflow、Caffe、Keras、MXNet等框架。
具體步驟:
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檔案目錄下cmd執行
python -m mmdnn.conversion._script.convertToIR -f keras -d keras_mlp -n mlp_model.json
5 瀏覽器開啟MMdnn Visualizer並選擇檔案