码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2018-05-25 13:23:47      阅读:194      评论:0      收藏:0      [点我收藏+]

标签:drawing   --   编写   imp   可视化   conf   utf-8   pyc   elf   

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)可视化

标签:drawing   --   编写   imp   可视化   conf   utf-8   pyc   elf   

原文地址:https://www.cnblogs.com/carle-09/p/9087101.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!