基于SE模块的SSD目标检测方法及计算机存储介质

作者：毛亮; 肖淳月; 刘三阳; 朱婷婷; 王祥雪; 谭焕新; 黄仝宇; 汪刚; 侯玉清; 刘双广

来源：2019-08-20, 中国, CN201910769868.0.

本发明提供一种基于SE模块的SSD目标检测方法及计算机存储介质,所述方法包括以下步骤：S1、获取需要进行目标识别的图片或者视频；S2、将卷积神经网络ResNet18的第一个卷积层替换为3×3卷积层,并在ResNet18的第一个和第二个残差块中添加SE模块,形成SE-ResNet18网络结构；S3、将SSD目标检测算法中的主干网络替换为所述SE-ResNet18网络结构,得到检测模型；S4、对所述检测模型进行针对小目标检测的训练,获得训练好的深度神经网络模型；S5、根据训练好的深度神经网络模型对所述图片或者视频的小目标进行检测,得到检测结果。根据本发明实施例的方法,既保证了一定的检测速度,又提升了小目标的检测精度,同时尽量降低了对模型大小的影响。

• 单位
  高新兴科技集团股份有限公司; 西安电子科技大学