关键词:
垃圾检测
双目视觉
深度学习
嵌入式系统
摘要:
随着人口数量的不断增加,在道路上随意丢弃垃圾的现象也愈发严重,而现有智能清洁设备大多适用于封闭场所,对检测设备的实时性要求不高,与城市道路工作环境差距较大,因此无法直接将其应用于道路垃圾检测。此外,智能环卫清洁设备所使用的环境感知方案多为激光雷达方案,对于部署量大的城市环卫工作来说,还存在应用成本较高的问题。针对上述问题分析,依托人工智能水平的进步,本文基于双目视觉系统及深度学习算法对环卫车辆进行升级,使其能够在较低成本的前提下,代替环卫工人实现对道路垃圾的检测和定位,提高道路垃圾检测效率。主要工作内容如下:(1)对比分析不同目标定位及深度学习算法,根据任务需要选定合适的算法及相机硬件,选择合适的深度学习算法,完成系统整体结构设计。(2)建立相机成像模型,根据双目相机立体匹配流程,首先对选定的双目相机,利用MATLAB进行相机标定,获取内外参数及畸变矩阵,建立成像模型;其次通过再现不同的图像预处理算法,选择合适的算法进行预处理,减小噪声干扰,提高成像质量;最后,对比分析不同立体匹配算法,根据选定算法完成双目相机立体匹配,实现目标精准定位。(3)收集道路垃圾图片并对其进行数据增强及标注,建立数据集。根据选定算法构建目标检测模型进行实验,与目前主流的目标检测模型进行对比,根据实验结果对模型进行改进。将主干网络替换为Efficient Net进行特征提取,同时减轻模型体积;利用CBAM双通道注意力机制提高对目标的关注度,减少背景信息的干扰,提高目标检测的精准度;建立多尺度特征融合结构,增强对不同尺度下特征图中目标信息的利用率,进一步提高检测精度;利用K-Means++算法进行聚类分析,得到匹配任务目标尺寸的Anchors尺寸,提高模型拟合速度;利用模型压缩算法对模型进行压缩,以提高其在嵌入式系统中的运行速度,提高检测效率。将改进后的模型与其他模型进行对比,证实改进算法的有效性。(4)对不同嵌入式平台进行对比,选择合适的平台及硬件,部署目标定位与检测模型并进行实验。实验结果表明,该系统在1000 mm至2000 mm范围能能够准确检测目标类别,且目标定位误差率在3%左右。此外,将优化完的轻量化算法与其他轻量化算法进行对比,结果表明本文算法的检测速度更快,能够满足任务实时性要求,对提高环卫效率具有实际意义。