【目标】采用pydot、pydot-ng、pydotplus和graphviz绘制模型拓扑结构。
【问题来源】
明明安装了pydot和graphviz,但是在jupyter notebook里运行的时候,还是没有结果,提示“You must install pydot (`pip install pydot`) and install graphviz (see instructions at https://graphviz.gitlab.io/download/) for plot_model to work."
【解决过程】
尝试了网上各种办法,修改文件啥的,还是不行。现在把我这边work的方案分享一下。
【解决方案】
环境:
python 3.8.19
tensorflow-gpu 2.10.0
pydot 2.0.0
pydot-ng 2.0.0
pydotplus 2.0.2
graphviz 0.20.3
安装好之后,配置环境变量很重要。新增一个用户变量, 把graphviz的路径填上去,注意是anaconda下面的graphviz的路径。
然后重启电脑(一定要),再在jupyternotebook里运行测试代码
from keras.applications import VGG16
from keras.utils import plot_model
from IPython.display import Image
conv_base=VGG16(weights='imagenet',
include_top=True)
# conv_base.summary()
plot_model(conv_base,show_shapes=True,to_file='VGG16.png', dpi=300)
# Image(filename='VGG16.png')
绘制结果如下,可以通过dpi的设置,得到不同清晰度的图。