面向车辆和行人检测的KM-SSD方法

• 作者: 郑钦浩 杨贞 杨振
• 来源期刊: 计算机与现代化
• 关键词: 单次检测器 车辆和行人检测 K-means++聚类算法 融合模块
• 年，卷（期）: 2021,（3）
• 所属期刊栏目: 人工智能|ARTIFICIAL INTELLIGENCE
• 页码范围: 51-56,62
• 页数: 7页
• 分类号: TP391.4
• 语种: 中文
• DOI: 10.3969/j.issn.1006-2475.2021.03.010

摘要

传统SSD改进方法在提升SSD目标检测精度的同时会降低其检测速度.针对这一问题,本文以SSD为基础,提出改进的KM-SSD方法.该方法首先利用K-means++聚类算法自适应学习先验框的宽高比例;然后设计高效特征融合模块实现高低层特征信息融合;最后本文在具有挑战性的KITTI数据集上对KM-SSD方法进行验证.实验结果表明,SSD的mAP为62.7％,平均检测时间为0.162 s;KM-SSD的mAP为69.8％,平均检测时间为0.133 s.因此,KM-SSD不仅提升了SSD在车辆和行人检测下的准确度,更是提高了SSD的检测速度,从而验证了本文所使用的K-means++聚类算法的有效性和特征融合方法的高效性.