Caffe Gadgets (1)-Visualize network structure
\quad recently studied Caffe, but as a former Windows depth user, it is a more accustomed visual interface. However, Caffe is certainly better under the Linux/os x system, because it is usually written in script that is played in the command line. So this is not intuitive, in order to be able to intuitively see the network structure, and not to see prototxt brain repair ... Visualization is important.
\quad Fortunately, the great gods who developed Caffe have considered the problem. In the root directory of Caffe, there is a Python folder, there is a draw_net.py, is the file we need.
\quad Next I picked up a random network structure. Suppose we are now in the path of this Python folder, at the command line, enter
Python draw_net.py. /models/bvlc_reference_caffenet/train_val.prototxt Vis.png
\quad I used a MacBook Pro to get an error on the first run
Pydot. Invocationexception:graphviz ' s executables not found
Check it out. I'm missing Graphviz. This software is dedicated to drawing, so it must be installed. If you have homebrew, that's easy.
Brew Install Graphviz
\quad runs again, it outputs the vis.png to the current path. Open to see:
\quad Unfortunately, for a layer of its own definition, there seems to be no way to visualize it at the moment (I tried to visualize the results of the FASTER-RCNN failed). Then I'll see how I can visualize the layer I've defined.