Caffe---Pycaffe进行网络结构(xxx.prototxt)可视化
Posted 烂笔头@WP
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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
以上是关于Caffe---Pycaffe进行网络结构(xxx.prototxt)可视化的主要内容,如果未能解决你的问题,请参考以下文章