1. 首页
  2. 编程语言
  3. Python
  4. 使用tensorflow实现VGG网络,训练mnist数据集方式

使用tensorflow实现VGG网络,训练mnist数据集方式

上传者: 2022-04-25 08:05:43上传 PDF文件 126.06 KB 热度 37次

VGG作为流行的几个模型之一,训练图形数据效果不错,在mnist数据集是常用的入门集数据,VGG层数非常多,如果严格按照规范来实现,并用来训练mnist数据集,会出现各种问题,如,经过16层卷积后,28*28*1的图片几乎无法进行。ILSVRC 2014的第二名是Karen Simonyan和 Andrew Zisserman实现的卷积神经网络,现在称其为VGGNet。它主要的贡献是展示出网络的深度是算法优良性能的关键部分。网络的结构非常一致,从头到尾全部使用的是3x3的卷积和2x2的汇聚。VGGNet不好的一点是它耗费更多计算资源,并且使用了更多的参数,导致更多的内存占用。其中绝大多数的参数都是来自于第一个全连接层。模型结构:本文在实现时候,尽量保存VGG原来模型结构,核心代码如下:卷积实现:最终效果:训练10000次后:结果如下:

用户评论