caffe系列(如何画网络结构图)

1.建立一个1.py的文件

#coding=utf-8
import caffe
import caffe.draw
from caffe.proto import caffe_pb2
from google.protobuf import text_format

# Set the parameters
input_net_proto_file = 'VDSR_net.prototxt' # net configuration file
output_image_file ='test_net.jpg' # the output image
rankdir ='LR' # net arrange method: LR、TB、RL

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

# Draw the net
print('Drawing net to %s' % output_image_file)
caffe.draw.draw_net_to_file(net, output_image_file, rankdir)
print('done...')
注意:

input_net_proto_file = 'VDSR_net.prototxt'  所要画的网络
output_image_file ='test_net.jpg'                  保存图片

2:python 1.py即可


若出现问题:1)No module named

解决方案如下:

在已经完成pycaffe的前提之下。

pip install pydot 

sudo apt-get -f install 

sudo apt-get -f install graphviz



你可能感兴趣的:(经验)