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')