卷积神经网络结构可视化

最新推荐文章于 2023-10-23 17:40:22 发布

W1510109901

最新推荐文章于 2023-10-23 17:40:22 发布

阅读量365

点赞数

分类专栏： LSTM 文章标签：神经网络

本文链接：https://blog.csdn.net/W1510109901/article/details/104043249

版权

LSTM 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

NN-SVG：
添加链接描述

ConvNetDraw：
添加链接描述

caffe可视化：
添加链接描述

from convnet_drawer import Model, Conv2D, MaxPooling2D, Flatten, Dense
from pptx_util import save_model_to_pptx
from matplotlib_util import save_model_to_file
model = Model(input_shape=(227, 227, 3))
model.add(Conv2D(96, (11, 11), (4, 4)))
model.add(MaxPooling2D((3, 3), strides=(2, 2)))
model.add(Conv2D(256, (5, 5), padding="same"))
model.add(MaxPooling2D((3, 3), strides=(2, 2)))
model.add(Conv2D(384, (3, 3), padding="same"))
model.add(Conv2D(384, (3, 3), padding="same"))
model.add(Conv2D(256, (3, 3), padding="same"))
model.add(MaxPooling2D((3, 3), strides=(2, 2)))
model.add(Flatten())
model.add(Dense(4096))
model.add(Dense(4096))
model.add(Dense(1000))
# save as svg file
model.save_fig("example.svg")
# save as pptx file
save_model_to_pptx(model, "example.pptx")
# save via matplotlib
save_model_to_file(model, "example.pdf")