[发明专利]车辆检测方法、装置、电子设备及存储介质

说明书

本发明实施例公开了一种车辆检测方法、装置、电子设备及存储介质，包括：获取交通摄像头采集的交通场景中的交通图像；复制第一数量的所述交通图像，每间隔第二数量的行或列对每一所述交通图像进行采样分割，以将每一所述交通图像分割为第三数量的图像块，根据预设的合并规则合并所述图像块，生成检测图像；提取所述检测图像中的特征信息，根据所述特征信息检测所述交通图像的车辆信息。通过对交通图像进行处理为检测图像后进行特征提取，可以有效提升模型对输入图像的感受精度，提高模型提取特征的全面性。再以较高全面性的特征作为依据检测交通图像中的车辆信息，从而提高对于交通图像中车辆检测的有效性。

技术领域

本发明涉及计算机应用技术领域，特别是涉及一种车辆检测方法、装置、电子设备及存储介质。

背景技术

交通部门在监控交通摄像头拍摄的画面时，需要对车辆类型进行检测，以协助工作人员对交通进行监管。

目前对交通画面进行车辆检测的方法通常是将交通画面直接输入到对应的模型中进行分类，根据分类结果确定画面中存在的车辆的信息。在现有技术的方案中，由于交通画面可能包含车辆的数量较多，因为相互遮挡等原因可能导致车辆识别并不全面；另一方面，画面中车辆尺寸相差会比较大(越远的车辆在画面中的尺寸越小)，直接利用交通画面进行识别的方案在车辆识别的准确度上无法得到保证。因此，亟需一种有效对交通画面进行车辆检测的方案。

发明内容

基于此，本发明提供一种车辆检测方法、装置、电子设备及存储介质，可以提高对图像中的车辆检测的有效性。

第一方面，提供一种车辆检测方法，包括：

获取交通摄像头采集的交通场景中的交通图像；

复制第一数量的所述交通图像，每间隔第二数量的行或列对每一所述交通图像进行采样分割，以将每一所述交通图像分割为第三数量的图像块，根据预设的合并规则合并所述图像块，生成检测图像；

提取所述检测图像中的特征信息，根据所述特征信息检测所述交通图像的车辆信息。

可选的，所述提取所述检测图像中的特征信息，根据所述特征信息检测所述交通图像的车辆信息，包括：

提取所述检测图像的特征信息；

根据所述特征信息进行特征重组，生成图像特征；

根据所述图像特征确定所述车辆信息。

可选的，所述根据所述特征信息进行特征重组，生成图像特征，包括：

对所述特征信息分别进行若干次上采样和下采样，得到若干上采样特征和下采样特征；

根据预设的连接规则依次连接所述上采样特征和下采样特征，生成所述图像特征。

可选的，所述上采样和下采样变换后的像素灰度值等于距所述变换后的像素最近的两个输入像素的灰度值的均值。

可选的，所述根据所述图像特征确定所述车辆信息，包括：

将所述交通图像分成预设数量的网格；

根据所述图像特征获取每个网格中至少两个目标边框的预测参数，其中，每个目标边框包含至少两个目标类别，预测参数包括目标边框中心点坐标、宽高和置信度，所述目标类别用于表示网格内容是否属于车辆；

过滤所有目标边框中置信度小于预设阈值的边框；

对保留的目标边框进行非极大值抑制处理，确定各目标类别置信度最高的目标边框的位置，作为所述车辆信息。

可选的，获取交通摄像头采集的交通场景中的交通图像之后，还包括；

获取所述交通图像的每个像素点的原始亮度，计算所述交通图像的每个像素点的平均亮度；