caffe利用python接口绘制网络结构图

#!/usr/bin/env python
#coding=utf-8

import sys
caffe_root = "D:\\Caffe\\caffe-windows\\Build\\x64\\Debug\\pycaffe"
sys.path.insert(0, caffe_root)

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

from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
from google.protobuf import text_format

print('finish import!')


# Set parameters
net_file = 'D:/1byav/MTCNN/test_only/codes/camera_demo/model/det1.prototxt'  
image_file ='D:/1byav/MTCNN/test_only/codes/camera_demo/model/cifa10.png'  
rankdir ='LR'  

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

print(net)
caffe.draw.draw_net_to_file(net, image_file, rankdir)
print('draw finish!')

参考： https://blog.csdn.net/qq_27432889/article/details/52457011