[发明专利]一种基于计算机视觉的渣土车未苫盖检测系统及检测方法

说明书

本发明公开了一种基于计算机视觉的渣土车未苫盖检测系统及检测方法，检测系统包括双模型训练模块、渣土车检测模块、渣土车斗区域检测模块、渣土车未苫盖判断模块、风险上报与记录模块，以及日志模块；本发明渣土车检测模型识别图像中的渣土车，并通过渣土车斗区域检测模型对车斗部分进行细分，计算各部分区域面积获得渣土的未覆盖率，若渣土未覆盖率大于给定的阈值，则将渣土车判定为未苫盖。本发明将目标检测技术与渣土车未苫盖相结合，采用双模型划分任务，并且对车斗部分进行细分，引入渣土未覆盖率对判断标准进行量化，提高识别的准确性，大大减少误报率。保证能够对渣土车辆做到有效的监管，减少环境污染和安全隐患。

技术领域

本发明属于计算机视觉领域，特别涉及一种基于计算机视觉的渣土车未苫盖检测系统及检测方法。

背景技术

为了维持和改善城市的居住环境，减少渣土车运输过程中的扬尘污染，不少相关部门采取了相关措施，除了加大了管理力量，也对相应的处罚力度予以增加；但由于渣土车行驶的时间、路线等无法掌控的因素，渣土车的监管还是存在着一定的难题。

依靠人力去检测渣土车未苫盖情况，存在非常大的不稳定性和主观性，难以全角度全时间的进行监控，同时将耗费巨大的人力物力成本。

目前，随着计算机视觉和深度学习的迅速发展，目标检测技术在传统领域的应用越来越广泛，通过卷积神经网络、深度置信网、神经网络等构建一个全方位多层次的学习系统，可以更好的发挥人工智能的准确性、高效性、便捷性。现有的渣土车未苫盖检测是将突破输入模型中直接输出是否苫盖的检测结果，误报率高。

发明内容

为解决上述技术问题，本发明提供了一种基于计算机视觉的渣土车未苫盖检测系统及检测方法，以达到提高识别准确性，减少误报率，提高对渣土车辆的监管，减少环境污染和安全隐患的目的。

为达到上述目的，本发明的技术方案如下：

一种基于计算机视觉的渣土车未苫盖检测系统，包括双模型训练模块、渣土车检测模块、渣土车斗区域检测模块、渣土车未苫盖判断模块、风险上报与记录模块，以及日志模块；

所述双模型训练模块负责渣土车检测模型和渣土车斗区域检测模型的训练；

所述视频帧获取模块负责通过轮询算法，连接对应的摄像头组，将不同摄像头的视频流进行运动检测，若视频画面运动，则按照一定时间间隔抽帧送入推理队列；

所述渣土车检测模块包含渣土车检测模型，所述渣土车检测模型基于改进的YOLOv5x模型，用于检测图片中的各种车辆，将检测到的渣土车辆用矩形框框出轮廓，从原图中分割发送给渣土车斗区域检测模块；

所述渣土车斗区域检测模块包含渣土车斗区域检测模型，所述渣土车斗区域检测模型基于改进的YOLOv5s模型，对渣土车车斗部分进行检测，得到渣土区域，空车斗区域和毡盖区域；

所述渣土车未苫盖判断模块负责根据车斗部分的检测结果计算渣土未覆盖率，从而判断渣土车是否苫盖；

所述风险上报与记录模块负责上传风险信息，保存风险图片；

所述日志模块负责记录系统运行当中的错误和警告信息，方便后期维护修改。

一种基于计算机视觉的渣土车未苫盖检测方法，采用如上所述的一种基于计算机视觉的渣土车未苫盖检测系统，包括如下步骤：

模型训练阶段：视频帧获取模块按照一定的时间间隔从视频流中收集图片，进行初始标注，制作初始数据集；双模型训练模块利用初始数据集训练渣土车检测模型，通过训练完成的渣土车检测模型将图片中的检测类别为渣土车的矩形框分割出来单独保存，组成渣土车斗区域检测模型的数据集；然后双模型训练模块利用渣土车斗区域检测模型的数据集训练渣土车斗区域检测模型；