python - 无法导入pydot

我正在尝试打印VGG16模型的模型摘要，并绘制模型并将其保存在.png文件中。

from keras.applications.vgg16 import VGG16
from keras.utils.vis_utils import plot_model

#Creating the object of VGG16 model
model=VGG16()
print(model.summary())
plot_model(model,to_file='vgg.png')

我还根据建议的here安装了以下软件包。

pip install pydot
pip install graphviz
pip install pydot-ng

这是在Windows10机器上的Python3.5上运行的。

下面是错误：

追溯（最近一次通话）：

File "<ipython-input-4-d5d9b64127c4>", line 1, in <module>
    runfile('C:/Users/Mohanakrishna/Desktop/Work/ObjectDetection.py',
    wdir='C:/Users/Mohanakrishna/Desktop/Work')

    File "c:\users\mohanakrishna\appdata\local\programs\python\python35\lib\site-packages\spyder\utils\site\sitecustomize.py", line 705, in runfile
    execfile(filename, namespace)

  File "c:\users\mohanakrishna\appdata\local\programs\python\python35\lib\site-packages\spyder\utils\site\sitecustomize.py", line 102, in execfile
    exec(compile(f.read(), filename, 'exec'), namespace)

  File "C:/Users/Mohanakrishna/Desktop/Work/ObjectDetection.py", line 14, in <module>
    plot_model(model,to_file='vgg.png')

  File "c:\users\mohanakrishna\appdata\local\programs\python\python35\lib\site-packages\keras\utils\vis_utils.py", line 135, in plot_model
    dot = model_to_dot(model, show_shapes, show_layer_names, rankdir)

  File "c:\users\mohanakrishna\appdata\local\programs\python\python35\lib\site-packages\keras\utils\vis_utils.py", line 56, in model_to_dot
    _check_pydot()

  File "c:\users\mohanakrishna\appdata\local\programs\python\python35\lib\site-packages\keras\utils\vis_utils.py", line 31, in _check_pydot
    raise ImportError('Failed to import pydot. You must install pydot'

ImportError: Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.

最佳答案

我有同样的问题。由于Windows 10上的任何原因，即使我将其添加到path中，也不想找到graphviz。所以最后我通过添加解决

os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/'

到我的项目。

关于python - 无法导入pydot，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/48507679/