如何对prototxt文件绘制网络的结构图

Posted 2023-05-10

参考技术A pip install pydotplus（需要安装pydotplus这个package）
在containter的bash中，利用：
python caffe/python/draw_net.py workspace/prettynet.prototxt workspace/prettynet.jpg11

workspace/prettynet.prototxt:prototxt文件
workspace/prettynet.jpg：目标ipg的名字（及位置）
或者是代码
# Load the module
import caffe
import caffe.draw
from caffe.proto import caffe_pb2
from google.protobuf import text_format

# Set the parameters
input_net_proto_file = 'test_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...')

train_val.prototxt文件和deploy.prototxt文件开头的区别

1.开头不同

对train_val.prototxt文件来说，开头部分定义训练和测试的网络及参数

对deploy.prototxt文件来说，开头部分定义实际运用场景的配置文件，其参数不定义数据来源，仅定义数据输入的格式大小

如下：

train_val.prototxt

deploy.prototxt