使用GraphViz画caffe网络结构图

时间:2023-03-09 19:48:17
使用GraphViz画caffe网络结构图

参考http://blog.****.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

显示结果

使用GraphViz画caffe网络结构图