科研之友
微信
新浪微博
Facebook
分享链接
摘要
针对现有识别算法使用的数据集过于简单、对真实场景中安检图像危险品的识别准确率低且容易导致误检和漏检的问题，基于渗透假设的类平衡分层求精提出了一种结合多层通道注意力机制和空间注意力机制的算法。首先，在安检图像分层建模的基础上在特征图中加入通道注意力机制，对不同通道特征赋予不同的权重。然后，加入空间注意力机制，在空间上对安检图像特有的颜色特征赋予不同的权重。最后，利用残差网络在安检图像不同层分别加入双注意力机制进行消融实验。实验结果表明，对固定两层同时添加双注意力机制后，网络对安检危险品的识别精度有显著提高，验证了多层注意力机制算法的有效性和鲁棒性。
-
单位电子信息工程学院; 河北工业大学
