Keras輸出網路結構圖

CG&AR Player發表於2020-11-21
Keras

pydot已經停止開發了，python3.5以上已經用不起來了。對策是：

pip uninstall pydot

pip install pydotplus

然後找到keras裡面的utils\vis_utils.py，把裡面的pydot的都替換成pydotplus

from keras.utils import plot_model
……
plot_model(modelname, to_file='./model.png')

vis_utils.py的位置可以先進入keras.utils，然後找到 from .vis_utils，再進入就可以了

相關文章