计算机工程

作者：张震, 李孟洲, 李浩方, 马军强

单位：郑州大学 电气工程学院, 郑州 450001

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

基金：国家重点研发计划“公共安全风险防控与应急技术装备”重点专项（2018YFC0824XXX）。

针对传统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%。

来源：2021年第7期

《计算机工程》期刊编辑部

查看计算机工程杂志2021年第7期