从零开始学caffe（六）：如何在win10下利用caffe绘制网络结构

之前两篇文章我们已经介绍了如何在win10中编译pycaffe，在这里，我们将介绍如何在win10下利用caffe绘制神经网络的结构。

利用draw_net.py

我们可以利用draw_net.py函数来绘制神经网络（需要确保已经按照前几篇文章配置好相关的环境）。draw_net.py共有三个参数，第一个参数为 “”-rankdir TB”。TB表示TOP到BOTTOM，从上到下对神经网路进行绘制，或者使用LR，表示从左到右对神经网络的结构进行绘制。第二个参数为网络结构描述文件的路径，第三个参数：网络图的输出路径。我们在CMD中根据要求对这三个参数进行配置如下：

\Users\MyPC>python E:\caffe-windows\python\draw_net.py --rankdir TB E:\caffe-windows\examples\mnist\lenet_train_test.prototxt E:\caffe-windows\examples\mnist\draw_net\mnist_TB.png
Drawing net to E:\caffe-windows\examples\mnist\draw_net\mnist_TB.png

在CMD中的运行结果为：

Drawing net to E:\caffe-windows\examples\mnist\draw_net\mnist_TB.png

表示神经网络结构已经存放在文件夹draw_net中。
打开draw_net，我们得到的结构如下：

利用在线画图软件

我们可以将网络的描述代码复制到网站，然后shift+enter即可得到相应的网络结构，这种方法比较简单，且可以很方便修改神经网络模型。