Ошибка при рисовании сети в Caffe

Question

Я пытаюсь нарисовать свою сеть в Caffe. Сеть определяется в expt/expt.prototxt, а желаемый путь изображения - expt/net.png. Когда я запускаю команду ./python/draw_net.py expt/expt.prototxt expt/net.png, я получаю следующее сообщение об ошибке:

Couldn't import dot_parser, loading of dot files will not be possible. 
Drawing net to expt/net.png 
Traceback (most recent call last): 
    File "./python/draw_net.py", line 44, in <module> 
    main() 
    File "./python/draw_net.py", line 40, in main 
    caffe.draw.draw_net_to_file(net, args.output_image_file, args.rankdir) 
    File "/home/pras/caffe/python/caffe/draw.py", line 165, in draw_net_to_file 
    fid.write(draw_net(caffe_net, rankdir, ext)) 
    File "/home/pras/caffe/python/caffe/draw.py", line 156, in draw_net 
    return get_pydot_graph(caffe_net, rankdir).create(format=ext) 
    File "/home/pras/anaconda/lib/python2.7/site-packages/pydot.py", line 1727, in create 
    'GraphViz\'s executables not found') 
pydot.InvocationException: GraphViz's executables not found 

Я бегу Caffe на Ubuntu 14.04 в режиме CPU с помощью интерфейса Anaconda Python. Любые идеи, почему эта ошибка может возникать и как ее исправить?

Answer 1

В журнале ошибок упоминается, что исполняемые файлы GraphViz не найдены. Итак, я сделал следующее:

1. Установленный GraphViz на Ubuntu: sudo apt-get install GraphViz.
2. Установленный GraphViz для Python: pip install GraphViz.

Я не уверен, требуется ли этап 2 или нет, но обязательно требуется шаг 1. После этого команда нарисовать сеть в Caffe работает как шарм!