Exception: "dot" not found in path.

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

root@aliyun:~/lisa-caffe-public/python# python draw_net.py ../examples/mnist/lenet_train_test.prototxt lenet5.pngDrawing net to lenet5.pngTraceback (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 graphvizpip install pydot

记录几个绘图的命令

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