caffe神经网络模型的绘图

最新推荐文章于 2024-04-26 18:36:03 发布

wuchao0508

最新推荐文章于 2024-04-26 18:36:03 发布

阅读量243

点赞数

本文链接：https://blog.csdn.net/wuchao0508/article/details/52919567

版权

本文介绍如何使用Python脚本draw_net.py将神经网络模型的prototxt文件转换为直观的图形化表示。首先需要安装GraphViz和pydot库，然后通过编译pycaffe来完成准备工作。最后，通过指定网络模型文件及输出方向参数，即可生成网络结构图。

摘要由CSDN通过智能技术生成

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

１、安装ＧraphViz

# 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/lib
PYTHON_LIB := $(ANACONDA_HOME)/lib

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

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

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

例：绘制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