首页 > 代码库 > caffe神经网络模型的绘图
caffe神经网络模型的绘图
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模型
# sudo python python/draw_net.py examples/mnist/lenet_train_test.prototxt netImage/lenet.png --rankdir=BT
caffe神经网络模型的绘图
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。