在 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

我在 Ubuntu 14.04 上 CPU 模式使用 Anaconda Python 界面运行ning Caffe。知道为什么会发生此错误以及如何纠正它吗？

Answer 1

错误日志提到找不到 GraphViz 的可执行文件。所以，我做了以下事情：

已在 Ubuntu 上安装 GraphViz：sudo apt-get install GraphViz。
为 Python 安装了 GraphViz：pip install GraphViz。

我不确定是否需要第 2 步，但肯定需要第 1 步。完成后，在 Caffe 中绘制网的命令就像一个魅力！

在 Caffe 中绘制网络时出错

Error while drawing net in Caffe

python

draw

ubuntu-14.04

deep-learning

caffe