【图像识别Bert版】运用AI解决样本过少的问题,提高图像识别效果——加入VGG16预训练基(卷积神经网络)

现实中,存在一些深度学习任务,某些样本的数据集样本量较少,面对这样的情况,我们该怎么做呢?
在这里插入图片描述

通常有如下几种方式:
1数据增强,通过图像裁剪,水平翻转等方式丰富有限的样本集。

2使用在大规模多种类数据集上训练好的AI模型,例如VGG16,
ResNet,作为预训练模型,使得小样本集在通用的特征提取器运作。

3由于数据集较少,在构建训练模型时,可以加入Dropout层,避免
样本局限带来的过拟合问题。

4在测试数据集上,进行数据匹配获取更多维度特征,填充标
签,丰富测试数据集质量,迭代引擎规则+预训练模型
训练新分类器后的分类模型。

在使用预训练模型时,请注意如下:
预训练模型的某个卷积层的通用性(以及可复用性)取决于该层在模型中的深度。模型中更靠近底部的层提取的是局部的、高度通用的特征图(比如视觉边缘、颜色和纹理),近顶部层则是越接近任务指向型的更加宏观整体的特定分类(比如识别“猫耳朵”或“狗眼睛”)。 因此,如果你的新数据集与原始模型训练的数据集有很大差异,那么最好只使用模型的前几层来做特征提取,而不是使用整个卷积基。

预训练模型的选择:
基于ImageNet 数据集上预训练得到的深度学习模型,以下都可以参考作为我们今天的预训练模型:

Xception
Inception
  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

AI深度学习算法实战代码解读

离下一篇原创,还差5个打赏

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值