参考http://blog.csdn.net/happynear/article/details/45440709
1. 安装pydot:
sudo pip install pydot
2. 安装Graphviz:
sudo apt-get install GraphViz
3.调用并显示
sudo python draw_net.py ../examples/mnist/lenet_train_test.prototxt 1.jpg
显示结果
时间: 2024-10-29 04:42:11