Caffe学习:使用pycaffe绘制网络结构
2015-09-10 13:41
423 查看
详见:/caffe/python/draw_net.py
导入相关库
参数配置
读取网络
绘制网络
注:如果提示缺少pydot库的话,执行如下命令安装:
导入相关库
import caffe import caffe.draw from caffe.proto import caffe_pb2 from 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
相关文章推荐
- IOS网络第一天 - 02SDWebImage
- C++用法的学习心得(要求包含示例,并反映出利用网络获取帮助的过程)
- IOS网络第一天-01基本的HTTP请求
- C#网络编程
- 同步与异步访问网络
- Java HTTPS客户端如何处理证书
- http://www.cnblogs.com/huangcong/archive/2010/06/14/1757957.html
- 神经网络图灵机(Neural Turing Machines, NTM)
- 用好虚拟网络编辑器
- 移动无线网络长连接 推送技术 已经服务器连接数多时解决策略
- 【转】TCP三次握手和四次握手协议
- HttpClient使用详解
- ios同步、异步请求网络
- OKHttp 使用
- 利用缓存的网络凭据入侵服务器
- HTTP常用状态代码
- netty 简单httpserver实现
- VM虚拟网络编辑器
- curl http_build_query踩过的坑
- 网络七层协议的形象说明