基于卷积神经网络的小样本图像识别方法 |
| |
引用本文: | 段萌,王功鹏,牛常勇. 基于卷积神经网络的小样本图像识别方法[J]. 计算机工程与设计, 2018, 0(1): 224-229 |
| |
作者姓名: | 段萌 王功鹏 牛常勇 |
| |
作者单位: | 郑州大学信息工程学院 |
| |
基金项目: | 国家自然科学基金项目(60901078);郑州市科技计划基金项目(10LJRC189) |
| |
摘 要: | 为提高仅包含少量训练样本的图像识别准确率,利用卷积神经网络作为图像的特征提取器,提出一种基于卷积神经网络的小样本图像识别方法。在原始小数据集中引入数据增强变换,扩充数据样本的范围;在此基础上将大规模数据集上的源预训练模型在目标小数据集上进行迁移训练,提取除最后全连接层之外的模型权重和图像特征;结合源预训练模型提取的特征,采用层冻结方法,微调目标小规模数据集上的卷积模型,得到最终分类识别结果。实验结果表明,该方法在小规模图像数据集的识别问题中具有较高的准确率和鲁棒性。
|
关 键 词: | 卷积神经网络 模型预训练 模型微调 迁移学习 小样本 |
Method of small sample size image recognition based on convolution neural network |
| |
Abstract: | |
| |
Keywords: | convolution neural network pre-trained model model fine-tune transfer learning small sample |
本文献已被 维普 等数据库收录! |
|