keras 模型可视化

from keras.utils import plot_model
from keras.models import load_model

model_path = "./data/yolo.h5"

if __name__ == '__main__':
    model = load_model(model_path)
    plot_model(model, to_file='model.png')

注意需要安装

pip install pydot

安装完pydot他又让你去安装graphviz，然后去官网安装一下，把bin目录加入环境变量。

然后我就可以了，（其实折腾了好多，一开始不行，有人说是graphviz安装到c盘了所以需要管理员权限，不啦不啦，然后一顿骚操作，就好了，其实啥也没干）

嗯，就这样吧