ubuntu16.04 绘制caffe net 网络结构_caffe u-net网络结构-优快云博客

本文链接：https://blog.youkuaiyun.com/swiftfake/article/details/79698959

本文介绍如何使用Python脚本draw_net.py绘制CaffeNet结构，包括必要的依赖安装步骤及常见问题解决方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我们在学习深度的网络的时候，为了能有更好的感性认识，可以使用draw_net.py将caffe net 结构绘制出来

安装python所需要的依赖

sudo apt-get install python-numpy python-scipy python-matplotlib ipython ipython-notebook python-pandas python-sympy python-nose

然后

sudo apt-get install graphviz #graphviz是一种很好用的画图工具
sudo su
pip install pydot #如果未安装pip，可以根据提示，自行安装

在安装完成后，我们发现还是不能绘图，那是因为我们使用了anaconda，所有我们需要将/usr/local/lib/python2.7/dist-packages 文件下的内容，全部复制到/home/XXX/anaconda2/lib/python2.7/site-packages（xxx为用户名）文件下。

测试绘制mnist的LeNet网络层结构命令：

cd caffe/python
python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet.png --rankdir=TB

效果如下：
这里写图片描述

常遇问题
这里写图片描述
此时如果使用

sudo pip install pydot

会提示没有pip这个命令，故希望读者能按博客里的建议安装步骤来，先切换到root用户，然后执行安装命令，如果提示没有pip，就按照提示安装pip就可以了。