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

权利要求书

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

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

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

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

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

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

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

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

2.一种基于计算机视觉的渣土车未苫盖检测方法，采用如权利要求1所述的一种基于计算机视觉的渣土车未苫盖检测系统，其特征在于，包括如下步骤：

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

检测阶段：视频帧获取模块按照一定的时间间隔从待检测的视频流中收集图片，输入到训练后的渣土车检测模块中，通过渣土车检测模型检测图片中出现的渣土车，并把轮廓用矩形框框出，将矩形框框出的渣土车从原图中分割出来并传给渣土车斗区域检测模块，渣土车斗区域检测模型检测渣土车的车斗区域，将车斗区域分成三个区域：渣土部分、空车斗部分和毡盖部分，对应检测的三个类别；渣土车未苫盖判断模块根据车斗部分的检测结果计算三个区域的面积，求得该渣土车的渣土未覆盖率，若渣土未覆盖率大于设定阈值，则认定该渣土车未苫盖；反之，则苫盖；风险上报与记录模块将未苫盖的渣土车的图片信息和检测车斗结果保存。

3.根据权利要求2所述的一种基于计算机视觉的渣土车未苫盖检测方法，其特征在于，模型训练阶段，渣土车检测模型的训练如下：

(1)从给定车流量较大路段的摄像头中，按照一定的时间间隔抽帧保存视频帧，筛选含有渣土车的图片；

(2)对图片中的车辆进行标注，得到m个基准框σ_i(x_i,y_i,w_i,h_i,l_i)，其中，i＝1,2,...,m，x_i,y_i,w_i,h_i,l_i五个分量分别为基准框左上角的横坐标、纵坐标，基准框的宽度、高度，以及标签，标签l_i＝0表示渣土车，l_i＝1表示大型货车，l_i＝2表示厢式货车，l_i＝3表示小型开放式货车，l_i＝4表示汽车，l_i＝5表示公交车，l_i＝6表示油罐车和混凝土车，l_i＝7表示其他车辆，包括挖掘机、平板车、装载动物的货车；

(3)对标注后的图片进行平移和旋转缩放来增加数据集；

(4)采用Mosaic进行数据增强；

(5)自适应锚框计算；

(6)自适应图片缩放；

(7)采用YOLOV5x网络进行训练，对检测出的车辆目标，用矩形实线框标出，构造二元交叉熵损失，包括边界框回归损失、置信度预测损失和类别预测损失三部分，进行反向传播。