weixin_52564360的博客

原创 初学：基于注意力机制的图像分类问题研究[附代码]

图像分类作为计算机视觉的一个重要研究领域，由传统机器学习到最近兴起的深度学习模型，不断地提高着分类的准确性与高效性。本文从数据集的匮乏问题出发，在原有数据集上进行了相应的扩充与图像增强，以此来提高模型的训练精度与泛化性能。针对原有的参考模型 IntelCNN分析相应复杂度、准确度等局限性，提出了本文的改进模型 NewCNN 与 基于注意力机制的模型 NewCNN_attention，由于归一化、注意力机制等操作的引入，模型在性能与效率方面均取得较好的效果。最后的消融研究也验证了模型中每个组件存在的必要性。