您的位置：首页 > 其它

Exception: "dot" not found in path.

2017-12-22 15:56 671 查看

我用pycaffe绘制神经网络的图时的问题

root@aliyun:~/lisa-caffe-public/python# python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet5.png
Drawing net to lenet5.png
Traceback (most recent call last):
File "draw_net.py", line 44, in <module>
main()
File "draw_net.py", line 40, in main
caffe.draw.draw_net_to_file(net, args.output_image_file, args.rankdir)
File "/root/lisa-caffe-public/python/caffe/draw.py", line 165, in draw_net_to_file
fid.write(draw_net(caffe_net, rankdir, ext))
File "/root/lisa-caffe-public/python/caffe/draw.py", line 156, in draw_net
return get_pydot_graph(caffe_net, rankdir).create(format=ext)
File "/usr/local/lib/python2.7/dist-packages/pydot.py", line 1883, in create
prog=prog))
Exception: "dot" not found in path.

解决方法

sudo apt-get install graphviz
pip install pydot

记录几个绘图的命令

python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet5.png
python draw_net.py ../examples/LRCN_activity_recognition/train_test_lstm_flow.prototxt train_test_lstm_flow.png
python draw_net.py ../examples/LRCN_activity_recognition/train_test_lstm_flow.prototxt train_test_lstm_flow_TB.png --rankdir TB

cc34

内容来自用户分享和网络整理，不保证内容的准确性，如有侵权内容，可联系管理员处理

标签：

相关文章推荐

新的分享

章节导航