1、安装pydot
sudo apt-get install python-pydot
2、安装graphviz
sudo apt-get instll graphviz
3、绘制
python draw_net.py ../examples/cifar10/cifar10_quick_train_test.prototxt cifa10.png
也可以在下面的网站上绘制网络结构图:
sudo apt-get install python-pydot
sudo apt-get instll graphviz
python draw_net.py ../examples/cifar10/cifar10_quick_train_test.prototxt cifa10.png