计算机应用研究

作者：毛志荣,都云程,肖诗斌,施水才,

单位：1.北京信息科技大学计算机学院,北京100101；2.拓尔思信息技术股份有限公司,北京100101；

关键词：注意力机制,深度学习,细粒度图像分类,多尺度,

针对细粒度图像分类问题提出了一种有效的算法以实现端到端的细粒度图像分类。ECA-Net中ECA（efficient channel attention）模块是一种性能优势显著的通道注意力机制，将其与经典网络ResNet-50进行融合构成新的基础卷积神经网络ResEca；通过物体级图像定位模块与部件级图像生成模块生成物体级图像和部件级图像，并结合原始图像作为网络的输入，构建以ResEca为基础的三支路网络模型Tb-ResEca-Net（three branch of ResEca network）。该算法在公有数据集CUB-200-2011、FGVC-aircraft和Stanford cars datasets上进行测试训练，分别取得了89.9%、95.1%和95.3%的准确率。实验结果表明，该算法相较于其他传统的细粒度分类算法具有较高的分类准确率以及较强的鲁棒性，是一种有效的细粒度图像分类方法。

来源：2021年第11期

《计算机应用研究》期刊编辑部

查看计算机应用研究杂志2021年第11期