基于自注意力的SSD图像目标检测算法 |
| |
引用本文: | 储岳中,黄勇,张学锋,刘恒.基于自注意力的SSD图像目标检测算法[J].华中科技大学学报(自然科学版),2020,48(9):70-75. |
| |
作者姓名: | 储岳中 黄勇 张学锋 刘恒 |
| |
作者单位: | 安徽工业大学计算机科学与技术学院,安徽 马鞍山 243002;安徽工业大学计算机科学与技术学院,安徽 马鞍山 243002;安徽工业大学计算机科学与技术学院,安徽 马鞍山 243002;安徽工业大学计算机科学与技术学院,安徽 马鞍山 243002 |
| |
基金项目: | 安徽省高等学校自然科学研究项目;国家自然科学基金 |
| |
摘 要: | 基于深度学习的方法,运用单次多框检测器(SSD)目标检测框架和自注意力机制,针对施工人员佩戴安全帽数据集进行神经网络训练.通过调整原始SSD目标检测框架中的参数,并向SSD目标检测框架中添加自注意力模块来计算特征图中像素点之间相互影响,以提高算法对目标检测的关注度,扩大卷积神经网络的感受野,从而提高目标检测的准确率.实验结果表明:改进算法在应对小目标检测以及目标之间的遮挡方面有很好的适应性,同时与其他检测算法相比,检测成功率有明显提高.
|
关 键 词: | 单次多框检测器(SSD) 卷积神经网络 自注意力 目标检测 安全帽检测 |
本文献已被 CNKI 万方数据 等数据库收录! |
|