轻量化SSD目标检测方法研究

作者:刘宽; 郎磊
来源:湖北民族大学学报(自然科学版), 2021, 39(04): 418-424.
DOI:10.13501/j.cnki.42-1908/n.2021.12.010

摘要

为了提高目标检测在嵌入式或移动端设备运行的可能性,基于SSD(Single Shot MultiBox Detector)框架,结合轻量化神经网络,构建一种轻量化SSD目标检测模型,称其为快速且精准的SSD(Fast and Accurate Single Shot Detector, FA-SSD).该方法采用轻量化卷积神经网络ESPNet作为基础网络,使用反卷积模块融合深浅层特征信息,并做轻量化处理,均衡模型尺寸和检测精度.实验结果表明,该方法相比原经典SSD算法具有更少的网络参数量和计算复杂度,在参数量上减少了47.3%,每秒处理图像帧数比经典SSD算法提升3.7倍.在VOC2007数据集中的测试平均精度均值(mAP)结果可以达到73.6%,和经典算法的结果相差无几,从而在保证检测精度的同时提高检测速度.

全文