Caffe学习：使用pycaffe绘制网络结构

详见：/caffe/Python/draw_net.py

导入相关库

import caffe
import caffe.draw
from caffe.proto import caffe_pb2from google.protobuf import text_format

参数配置

# 网络结构文件
input_net_proto_file = 'caffe_net.prototxt'
# 输出图片文件
output_image_file = 'net_pic.jpg'
# 网络结构排列方式：LR、TB、RL等
rankdir = 'LR'

读取网络

net = caffe_pb2.NetParameter()
text_format.Merge(open(input_net_proto_file).read(), net)

绘制网络

print('Drawing net to %s' % output_image_file)
caffe.draw.draw_net_to_file(net, output_image_file, rankdir)
print('done...')

注：如果提示缺少pydot库的话，执行如下命令安装：

 sudo apt-get install python-pydot

Caffe学习：使用pycaffe绘制网络结构相关推荐

Caffe学习系列(19): 绘制loss和accuracy曲线
转载自: Caffe学习系列(19): 绘制loss和accuracy曲线 - denny402 - 博客园 http://www.cnblogs.com/denny402/p/5110204.htm ...
Caffe学习系列(18): 绘制网络模型
转载自: Caffe学习系列(18): 绘制网络模型 - denny402 - 博客园 http://www.cnblogs.com/denny402/p/5106764.html python/dr ...
caffe学习日记--lesson7:caffemodel可视化的两种方法
在Caffe中,目前有两种可视化prototxt格式网络结构的方法: 使用Netscope在线可视化使用Caffe提供的draw_net.py 本文将就这两种方法加以介绍 Netscope:支持Ca ...
Caffe 学习系列
学习列表: Google protocol buffer在windows下的编译 caffe windows 学习第一步:编译和安装(vs2012+win 64) caffe windows学习:第一 ...
Caffe学习系列(13)：数据可视化环境（python接口)配置
原文有更新: Caffe学习系列(13):数据可视化环境(python接口)配置 - denny402 - 博客园 http://www.cnblogs.com/denny402/p/5088399. ...
Caffe学习笔记4图像特征进行可视化
Caffe学习笔记4图像特征进行可视化本文为原创作品,未经本人同意,禁止转载,禁止用于商业用途!本人对博客使用拥有最终解释权欢迎关注我的博客:http://blog.csdn.net/hit201 ...
Caffe学习笔记2
Caffe学习笔记2-用一个预训练模型提取特征本文为原创作品,未经本人同意,禁止转载,禁止用于商业用途!本人对博客使用拥有最终解释权欢迎关注我的博客:http://blog.csdn.net/hi ...
Caffe学习笔记2--Ubuntu 14.04 64bit 安装Caffe（GPU版本）
0.检查配置 1. VMWare上运行的Ubuntu,并不能支持真实的GPU(除了特定版本的VMWare和特定的GPU,要求条件严格,所以我在VMWare上搭建好了Caffe环境后,又重新在Windo ...
深度学习21天实战caffe学习笔记《3 ：准备Caffe环境》
准备Caffe环境 [如果是其他环境下的配置就请绕道喽,我也没有专门去试一试各个环境下的配置,请谅解~] 官网 http://caffe.berkeleyvision.org/installation ...
Caffe学习系列(8)：solver优化方法
Caffe学习系列(8):solver优化方法上文提到,到目前为止,caffe总共提供了六种优化方法: Stochastic Gradient Descent (type: "SGD&qu ...

Caffe学习：使用pycaffe绘制网络结构

Caffe学习：使用pycaffe绘制网络结构相关推荐

最新文章

热门文章