caffe中网络结构的可视化

在caffe中，网络结构文件（prototxt）的可视化，有两种方法。一种是Caffe里的工具 draw_net.py   (该文件目录 $(CAFFE_ROOT)/python/ )，还有一种是在线网页生成。

第一种方法使用如下：

usage: draw_net.py [-h] [--rankdir RANKDIR] [--phase PHASE]     input_net_proto_file     output_image_file

sudo  python python/draw_net.py  *.prototxt  *.png  --rankdir=BT（或者，TB，LR，RL）

 

第二种方法使用如下：

http://ethereon.github.io/netscope/#/editor    

或者  https://dgschwend.github.io/netscope/#/editor 

 

第三种方法使用如下：

https://lutzroeder.github.io/netron/

该工具开源地址：https://github.com/lutzroeder/netron

 

根据个人使用情况来说，推荐使用第二种或者第三种在线网页生成可视化网络结构，非常简便。

目前来说第三种支持更多的框架。