|
| 融合显著性和非局部模块的细粒度图像分类算法 |
| Fine-grained image classification algorithm combining saliency and non-local module |
| 投稿时间:2023-12-05 |
| DOI:10.3969/j.issn.1005-5630.202312050131 |
| 中文关键词: 细粒度图像分类 显著性 数据增强 深度学习 |
| 英文关键词:fine-grained image classification saliency data augmentation deep learning |
| 基金项目: |
|
| 摘要点击次数: 1964 |
| 全文下载次数: 871 |
| 中文摘要: |
| 针对细粒度图像分类任务中存在鉴别性特征获取不准确和训练数据利用不充分的问题,提出了融合显著性和非局部模块的细粒度图像分类算法。通过裁剪4张训练图像的显著性区域将其拼接为1张训练图像来丰富和增强训练数据,并将非局部模块嵌入到ResNet-50模型高维特征层中的瓶颈模块中来联系所增强图像的4个显著性区域,以加强模型对全局上下文信息的关注。在Stanford Cars和CUB-200-2011数据集上,Top-1分类准确率分别达到了94.01%和85.97%。与数据增强方法和细粒度图像分类算法相比,本方法性能更优。 |
| 英文摘要: |
| Aiming at the problems of inaccurate discriminative feature acquisition and insufficient use of training data in fine-grained image classification tasks, a fine-grained image classification algorithm combining significance and non-local modules is proposed. By clipping the significance region of four training images and stitching them into one training image, the training data could be enriched and enhanced. Moreover, the non-local module was embedded into the bottleneck module in the high-dimensional feature layer of the ResNet-50 model to connect the four salience regions of the enhanced image, which strengthened the model's attention to the global context information. On the Stanford Cars and CUB-200-2011 data sets, the accuracy of Top-1 classification was 94.01% and 85.97%, respectively. This method performs better than compared data augmentation methods and fine-grained image classification algorithms. |
| HTML 查看全文 查看/发表评论 下载PDF阅读器 |
| 关闭 |