改进SSD算法及其在地铁安检中的应用

摘要：

针对传统SSD算法在检测小目标时容易漏检且检测精度不高的问题,提出一种改进算法.对SSD算法中各尺度特征进行尺寸大小不变的卷积操作,将卷积前后对应的特征进行轻量级网络融合,从而生成新的金字塔特征层,并加入基于残差模块的检测单元避免增加网络模型容量和运算复杂度,同时增强对小尺度目标的检测能力.基于PASCAL-VOC2007小目标数据集的实验结果表明,与传统SSD、YOLOv3、Faster RCNN等算法相比,在PASCAL-VOC2007小目标数据集中,该算法的mAP指标较传统SSD算法提高8.5％,较Faster RCNN算法提高3.9％,较YOLOv3提高2％,FPS达到83 frame/s,其检测地铁安检图片的mAP达到77.8％.

关键词：SSD算法;网络融合;金字塔特征层;残差模块;检测单元;目标检测

所属期刊栏目：47

分类号：TP391(计算技术、计算机技术)

资助基金：国家重点研发计划"公共安全风险防控与应急技术装备"重点专项2018YFC0824XXX

在线出版日期：2021-09-15（万方平台首次上网日期，不代表论文的发表时间）

页数：共7页

页码：314-320