keras繪製網絡結構圖

keras可以方便畫出網絡結構圖,但在使用過程中有兩個庫需要先安裝 pydot 和graphviz

os: unbantu

pydot使用pip很容易安裝
graphviz直接用pip裝沒問題,但是使用時候遇到環境變量問題。這兒需要將graphviz/bin添加到環境變量,但是我裝完發現graphviz文件下無bin文件。。。

另尋他路,apt-get install graphviz直接安裝成功並且使用沒問題了~

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "  "
from keras.applications.resnet50 import ResNet50
model = ResNet50(weights='imagenet')
from keras.utils import plot_model
plot_model(model, to_file='model.png')
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章