Win7 caffe use notes - draw_net.py draw caffe network

To provide a benefit first, Netscope's online visualization tool http://ethereon.github.io/netscope/#/editor is only used to view the network structure. If you remove #/editor, you can also see alexnet, vgg16, googlenet and other networks structure.

Follow the steps under ubuntu to test and write the process successfully:


1. The microsoft for caffe version is used, and the pycaffe interface uses Anaconda2, giving two good configuration blogs

http://m.blog.csdn.net/article/details?id=51355143

http://blog.csdn.net/tina_ttl/article/details/51722983


2. Write a bat file. The caffe root directory /python/draw_net.py is called under ubuntu, and the root directory /Build/x64/Release/pycaffe/draw_net.py is downgraded under windows. 

Three parameters for execution The
first parameter: the prototxt file of the network model The
second parameter: the saved image path and name
The third parameter: --rankdir=x , x has four options, namely LR, RL, TB , bt. Used to indicate the direction of the network, from left to right, from right to left, from top to small, and from bottom to top. Defaults to LR.


G:
cd G:\caffe-master
SET GLOG_logtostderr=1

python  Build\x64\Release\pycaffe\draw_net.py  .\examples\Planthopper\draw_net.prototxt  examples\Planthopper\net\Planthopper_net.png
pause

There will be an error No module named google.protobuf


3. Add the python installation directory to the path system variable

Add the python directory \Scripts to the path system variable

cmd command pip install protobuf pydot

Install Graphviz  http://www.graphviz.org/Download_windows.php

Add the bin path to the path system variable


4. The retest was successful


Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=325608044&siteId=291194637