基于空间和ECA注意力机制的废旧手机型号识别研究

作者：余光海; 付勇刚; 邓梅玲

来源：日用电器, 2022, (11): 71-75.

DOI：10.3969/j.issn.1673-6079.2022.11.015

• 手机型号识别
• 残差网络结构
• 注意力机制
• 废旧手机回收

针对废旧手机的分类回收难以快速精准识别的问题,本文基于残差网络模型Resnet 34进行改进,让模型能自适应地关注到感兴趣的特征和定位到具有判别性的区域,从而提高手机型号识别的准确率。首先,在Resnet 34模型的Layer 4上添加ECA注意力机制,并实验将ECA通道注意力机制与空间注意力机制结合,组成新的Eca-S-Resnet 34注意力机制。然后对网络模型进行训练,基于验证集测试网络模型预测的准确率,最后导入照片测试手机型号识别速度与准确率。实验证明在模型的Layer 4上添加的ECA注意力机制与Eca-S-Resnet 34注意力机制均可以提高网络训练速度与模型的分类准确率。

• 单位
  合肥工业大学; 中国电器科学研究院股份有限公司