1、安装graphViz
sudo apt-get install graphviz
2、安装pydot
sudo pip install pydot
3、执行python中的一个绘图脚本,第一个参数是网络模型文件,第二个参数是保存图片的路径及名字,
第三个有4个选项,有LR、RL、TB、BT。分别是从左到右,从右到左,从上到下,从下到上,默认为LR
sudo python /home/yourname/caffe/python/draw_net.py\
/home/yourname/caffe/examples/mnist/lenet_train_test.prototxt\
/home/wangsen/caffe_case/lenet.png --rankdir=BT
caffe绘制网络图
最新推荐文章于 2020-06-27 20:26:30 发布