VGG16训练时间长,Accuracy 比率固定不变问题

来源:4-4 VGG-ResNet实战(1)

慕沐7111664

2020-12-20

环境
CPU R7-4800H
Memery 32G
Tensorflow 2.3 (无GPU)

使用 https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz 的鲜花图片集作为训练和验证集。依据教学自行搭建了VGG16网络,并使用了Dropout方法。训练10次发现如下2现象:

  1. 每次训练时间大约在 740s~780s,和教学演示速度差距很大。
  2. 训练10次发现 accuracy 比率一直固定在 0.18多,并不会出现 accuracy 增长的情况。转化使用5层的网络发现 accuracy 比率却可以达到 0.88

问题:
a. VGG在训练较大的图片时一直会这么慢吗,是否有相应办法可以加速训练?
b. VGG的 accuracy 比率为和会固定 0.1816的数值上,是否是网络搭建有问题?
c. 5层的卷积网络在有限的图片识别的分类上结果能够达到很高的 accuracy 比率,是否可以认为普通5层网络已经足够进行物体分类识别?

vgg16和5层卷积网络对比
图一: 自行构建vgg16
图片描述
图二: vgg16 model summary层次。
图片描述
图三: vgg16模型训练的 accuracy 比率
图片描述
图四:对比,自行搭建5层网络
图片描述

图五: 对比,五层网络 Model summary
图片描述
图六: 对比,5层网络训练结果
图片描述

写回答

1回答

正十七

2021-01-07

> a. VGG在训练较大的图片时一直会这么慢吗,是否有相应办法可以加速训练?

大图片带来的时间是平方级的变化,会慢很多,建议用GPU。

> b. VGG的 accuracy 比率为和会固定 0.1816的数值上,是否是网络搭建有问题?

网络的搭建感觉不是问题,可能在于初始化,optimizer, learning rate上。

> c. 5层的卷积网络在有限的图片识别的分类上结果能够达到很高的 accuracy 比率,是否可以认为普通5层网络已经足够进行物体分类识别?

不可以,泛化能力肯定是随着模型越大,数据集越大而逐步提升的。

0
0

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学,深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

查看课程