基于自注意力的SSD图像目标检测算法

作者：储岳中; 黄勇; 张学锋; 刘恒

来源：华中科技大学学报(自然科学版)科技大学, 2020, 48(09): 70-75.

DOI：10.13245/j.hust.200912

• 单次多框检测器(SSD)
• 卷积神经网络
• 自注意力
• 目标检测
• 安全帽检测

基于深度学习的方法,运用单次多框检测器(SSD)目标检测框架和自注意力机制,针对施工人员佩戴安全帽数据集进行神经网络训练.通过调整原始SSD目标检测框架中的参数,并向SSD目标检测框架中添加自注意力模块来计算特征图中像素点之间相互影响,以提高算法对目标检测的关注度,扩大卷积神经网络的感受野,从而提高目标检测的准确率.实验结果表明:改进算法在应对小目标检测以及目标之间的遮挡方面有很好的适应性,同时与其他检测算法相比,检测成功率有明显提高.

• 单位
  安徽工业大学