博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
caffe神经网络模型的绘图
阅读量:5059 次
发布时间:2019-06-12

本文共 1306 字,大约阅读时间需要 4 分钟。

Python/draw_net.py, 这个文件,就是用来绘制网络模型的。也就是将网络模型由prototxt变成一张图片。

1、安装GraphViz

# sudo apt-get install GraphViz

也可以去官网下载,手动安装

 

2 、安装pydot

 

sudo apt-get install sudo apt-get install 或者下边的方法编译。
# sudo pip install pydot

 

3.编译pycaffe

#sudo make pycaffe

编译之前可能需要配置makefile.config文件,计算机需要安装anaconda,并加入路径到配置文件中去。

# Anaconda Python distribution is quite popular. Include path:# Verify anaconda location, sometimes it's in root. ANACONDA_HOME := $(HOME)/anaconda2 PYTHON_INCLUDE := $(ANACONDA_HOME)/include \         $(ANACONDA_HOME)/include/python2.7 \         $(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include \# Uncomment to use Python 3 (default is Python 2)# PYTHON_LIBRARIES := boost_python3 python3.5m# PYTHON_INCLUDE := /usr/include/python3.5m \#                 /usr/lib/python3.5/dist-packages/numpy/core/include# We need to be able to find libpythonX.X.so or .dylib.PYTHON_LIB := /usr/libPYTHON_LIB := $(ANACONDA_HOME)/lib

 

 

安装好了,就可以调用脚本来绘制图片了

第一个参数:网络模型的prototxt文件

第二个参数:--rankdir=x , x 有四种选项,分别是LR, RL, TB, BT 。用来表示网络的方向,分别是从左到右,从右到左,从上到小,从下到上。默认为LR。

例:绘制Lenet模型

进入caffe/python目录

# sudo /home/wuchao/anaconda2/bin/python draw_net.py /home/wuchao/caffe/models/bvlc_alexnet/train_val.prototxt  /home/wuchao/caffe/MyFiles/alexnet.png --rankdir=BT

转载于:https://www.cnblogs.com/wuchaodzxx/p/5885186.html

你可能感兴趣的文章
HEVC播放器出炉,迅雷看看支持H.265
查看>>
[置顶] Android仿人人客户端(v5.7.1)——人人授权访问界面
查看>>
Eclipse 调试的时候Tomcat报错启动不了
查看>>
【安卓5】高级控件——拖动条SeekBar
查看>>
ES6内置方法find 和 filter的区别在哪
查看>>
Android入门之文件系统操作(二)文件操作相关指令
查看>>
Android实现 ScrollView + ListView无滚动条滚动
查看>>
java学习笔记之String类
查看>>
UVA 11082 Matrix Decompressing 矩阵解压(最大流,经典)
查看>>
jdk从1.8降到jdk1.7失败
查看>>
硬件笔记之Thinkpad T470P更换2K屏幕
查看>>
【知识库】-数据库_MySQL 的七种 join
查看>>
iOS开发——缩放图片
查看>>
HTTP之URL的快捷方式
查看>>
满世界都是图论
查看>>
配置链路聚合中极小错误——失之毫厘谬以千里
查看>>
代码整洁
查看>>
蓝桥杯-分小组-java
查看>>
Java基础--面向对象编程1(类与对象)
查看>>
Android Toast
查看>>