解决(‘You must install pydot (`pip install pydot`) and install graphviz (see…) ‘, ‘for plot_model..

目录

  • 一、报错提示
  • 二、解决方案(共四步)
  • 2.1 安装pydot
  • 2.2 下载并安装graphviz包
  • 2.3 配置环境变量(graphviz)
  • 2.4 调用
  • 三、结果展示
  • 一、报错提示

    调用keras.utils.plot_model报错提示:
    ('You must install pydot (pip install pydot) and install graphviz (see instructions at https://graphviz.gitlab.io/download/) ', ‘for plot_model/model_to_dot to work.’)

    因为tensoerflow2.0以上兼容了keras,只是产生图而已,所以不必在这里使用tf.keras,再者使用tf.keras也会报错,tf的库里没有这个函数。

    二、解决方案(共四步)

    2.1 安装pydot

    按照提示,安装pydot

    pip install pydot
    

    2.2 下载并安装graphviz包

    踩坑:千万别pip install graphviz。因为测试过无效。
    正确做法是从官网
    找到如下图所示的Win10 64位版本的graphviz进行下载,然后安装


    下一步就好了

    2.3 配置环境变量(graphviz)

    进入到安装的bin目录下


    变量名:别与系统的专用词冲突即可
    变量值:路径,我这里是D:\Graphviz\bin (仅供参考)

    2.4 调用

    添加环境变量后,引入包

    from keras.utils.vis_utils import plot_model
    

    模型编译后调用plot_model函数(见下图)
    (to_file中设置产生图片的路径,下面这种方式表示保存到当前文件下)

    keras.utils.plot_model(model, to_file='graph.png', show_shapes=True)
    

    三、结果展示

    模型编译后运行就会产生对应的可视化结构


    由于比较长,只截取部分

    来源:sinysama

    物联沃分享整理
    物联沃-IOTWORD物联网 » 解决(‘You must install pydot (`pip install pydot`) and install graphviz (see…) ‘, ‘for plot_model..

    发表评论