环境
- ubuntu
- pycaffe
步骤
1.安装pydot(可使用pip list命令 查看是否已安装)
pip install pydot
2.安装graphviz
sudo apt-get install graphviz
3.绘图
进入caffe-master/python目录下运行:
python draw_net.py ../examples/cifar10/ cifar10_quick_train_test.prototxt cifa10.png
pip install pydot
sudo apt-get install graphviz
进入caffe-master/python目录下运行:
python draw_net.py ../examples/cifar10/ cifar10_quick_train_test.prototxt cifa10.png