Caffe---Pycaffe进行网络结构(xxx.prototxt)可视化

Pycaffe---进行网络结构(xxx.prototxt)可视化

解决网络结构(xxx.prototxt)可视化，还可以借助python接口，编写一个类似如下的pycaffe_draw_net.py:

# -*- coding: utf-8 -*-
#!/usr/bin/env python 
 
import caffe
import caffe.draw
from caffe.proto import caffe_pb2
from google.protobuf import text_format  
 
# Set the parameters
input_net_file = '/home/wp/caffe/myself/road/prototxt_files/train_val.prototxt' # net configuration file
output_image_file ='/home/wp/caffe/myself/road/Log/road_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_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...')

# cd caffe

# python  ./xx/xx/pycaffe_draw_net.py