摘要
为了提高道路裂缝检测的速度和精度,同时适应嵌入式设备部署要求,提出一种改进的轻量级SSD道路裂缝检测算法。通过建立VOC格式的城市道路裂缝数据集,使用SSD为基础目标检测网络,将主干特征提取网络由VGG改进为轻量级的MobileNet网络,并通过特征金字塔进行融合构建出轻量级的道路裂缝检测算法,训练中采用迁移学习和数据扩增技术,在建立的数据集上将改进的算法分别与YOLO V3算法、SSD算法进行比较。结果表明,本文算法的MAP为91.29,比SSD算法提高2.05%,比YOLO V3算法提高8.16%。本文算法的FPS为38.1比YOLO V3算法和SSD算法分别提高1.1倍和1.88倍。改进的算法在检测精度方面相比SSD算法和YOLO V3算法有微弱优势,在检测速度方面优势明显,同时由于网络的轻量化,更方便在嵌入式设备上部署。
- 单位