解决ImportError: Failed to import pydot,You must install pydot and graphviz for `pydotprint` to work
问题代码如下:
....
import os
from tensorflow.keras.utils import plot_model
baseModel = VGG16(weights="imagenet", include_top=False, input_tensor=Input(shape=(224, 224, 3)))
for (i, layer) in enumerate(baseModel.layers):
print("[INFO] {}\t{}".format(i, layer.__class__.__name__))
headModel = FCHeadNet.build(baseModel, 17, 256)
model = Model(inputs=baseModel.input, outputs=headModel)
for (i, layer) in enumerate(model.layers):
print("[INFO] {}\t{}".format(i, layer.__class__.__name__))
plot_model(model, to_file="/Users/liushanlin/PycharmProjects/DLstudy/plotModel/fineTuneNet.png", show_shapes=True)
报错:
ImportError: Failed to import pydot. You must install pydot and graphviz for `pydotprint` to work.
然后我用pip指令安装了graphviz 和pydot 发现仍然报以上错误。
于是找到了以下方法:
-
打开终端(command+space然后输入Terminal后按回车)
-
输入以下命令并回车
ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)" 2> /dev/null
如果需要输入密码则输入密码,需要按回车就回车,总之按照要求来,然后等待执行完成。
- 然后执行
brew install graphviz
安装完成,发现可以用了。
创作不易,有用请顺手点赞,谢谢!