Caffe学习：使用pycaffe绘制网络结构

详见：/caffe/python/draw_net.py

• 导入相关库

import caffe
import caffe.draw
from caffe.proto import caffe_pb2

from google.protobuf import text_format

• 参数配置

# 网络结构文件
input_net_proto_file = 'caffe_net.prototxt'
# 输出图片文件
output_image_file = 'net_pic.jpg'
# 网络结构排列方式：LR、TB、RL等
rankdir = 'LR'

• 读取网络

net = caffe_pb2.NetParameter()
text_format.Merge(open(input_net_proto_file).read(), net)

• 绘制网络

print('Drawing net to %s' % output_image_file)
caffe.draw.draw_net_to_file(net, output_image_file, rankdir)
print('done...')

• 注：如果提示缺少pydot库的话，执行如下命令安装：

 sudo apt-get install python-pydot