keras 修改vgg19 对mnist 分类

最新推荐文章于 2023-01-25 09:10:52 发布

阿豪boy

最新推荐文章于 2023-01-25 09:10:52 发布

阅读量366

点赞数

文章标签：网络深度学习 python tensorflow 神经网络

本文链接：https://blog.csdn.net/qq_35516360/article/details/122066074

版权

原文链接: keras 修改vgg19 对mnist 分类

上一篇: python 变量大小，进程和内存信息

下一篇: TensorFlow 简单梯度下降可视化

修改后的网络模型，在使用了最小size 48X48时训练参数还是很多。。。。

48
Total params: 38,947,914
Trainable params: 18,923,530
Non-trainable params: 20,024,384

224
Total params: 139,611,210
Trainable params: 119,586,826
Non-trainable params: 20,024,384

网络各层信息

_________________________________________________________________
Layer (type)                 Output Shape              Param #
=================================================================
input_1 (InputLayer)         (None, 48, 48, 3)         0
_________________________________________________________________
block1_conv1 (Conv2D)        (None, 48, 48, 64)        1792
_________________________________________________________________
block1_conv2 (Conv2D)        (None, 48, 48, 64)        36928
_________________________________________________________________
block1_pool (MaxPooling2D)   (None, 24, 24, 64)        0
_________________________________________________________________