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文件目录下cmd执行
python -m mmdnn.conversion._script.convertToIR -f keras -d keras_mlp -n mlp_model.json
5 浏览器打开MMdnn Visualizer并选择文件