[发明专利]基于多尺度YOLOv5嵌套的车辆车牌检测定位方法

说明书

一种基于多尺度YOLOv5嵌套的车辆车牌检测定位方法，步骤包括：1)通过视频流，截取单帧图片；2)对图像进行HSV增强；3)训练大尺度YOLOv5网络作为车辆检测网络；3)训练小尺度YOLOv5网络作为车牌检测网络；4)使用大尺度YOLOv5车辆检测网络检测车辆，对步骤2)输入图像进行检测，得到车辆检测ROI；6)用车辆检测ROI对原始输入图像进行裁剪，得到图像中的车辆部分，再采用步骤2)的图像处理方法对车辆图片逐个处理；7)使用小尺度YOLOv5车辆检测网络检测车牌，对步骤6)得到的车辆图片逐个检测，得到每个车辆的车牌检测坐标。本方法单独对每个车辆放大后的检测，实现更高的精度。

技术领域

本发明涉及一种车辆和车牌的检测定位方法，主要是基于人工智能技术领域中的深度学习技术。

技术背景

车辆车牌检测是智慧交通领域的一个重要研究分支，也是车牌识别的基础，其主要任务是将公路监控视频车辆检测出来再逐个对单一车辆的车牌进行检测和定位。该任务在公路车辆行为监控场景有着广泛的应用需求。面对公路车辆的不文明行为，该任务结合车牌识别可以直接锁定车辆的车牌信息，对于公路监控视频检查可以节约监督人员人力成本，提高监督效率，覆盖更广的监督范围，加强安全管控。

发明内容

针对现有的技术问题，本发明的目的在于实现对于公路监控视频中车辆车牌的检测和定位，本发明旨在提出一种基于多尺度YOLOv5嵌套的深度学习检测算法，利用它能直接检测定位出公路视频在车牌的位置。

本发明是一种基于多尺度YOLOv5嵌套的车辆车牌检测定位方法，步骤包括：

1)通过视频流，截取单帧图片；

2)对图像进行HSV增强，增强后的图片与原始图片的比例一致；

3)训练大尺度YOLOv5网络作为车辆检测网络：

3)训练小尺度YOLOv5网络作为车牌检测网络；

4)使用大尺度YOLOv5车辆检测网络检测车辆，对步骤2)输入图像进行检测，得到车辆检测ROI；

6)用车辆检测ROI对原始输入图像进行裁剪，得到图像中的车辆部分，再采用步骤2)的图像处理方法对车辆图片逐个处理；

7)使用小尺度YOLOv5车辆检测网络检测车牌，对步骤6)得到的车辆图片逐个检测，得到每个车辆的车牌检测坐标。

本方法通过多尺度的YOLOv5嵌套模型实现车辆和车牌的检测，采用大尺度的YOLOv5模型对公路监控视频检测车辆，定位到车辆区域后，将车辆图片放大后送入小尺度YOLOv5模型检测车牌区域，最终实现车辆和车牌的检测定位。

该算法相比传统的从视频中直接检测车牌，定位车辆后单独对每个车辆进行放大后的检测，可以实现更高的精度。

本发明提出的车辆车牌检测方法是基于人工智能技术领域中的深度学习技术，结合大数据，在实施时候，使用大量的公路车辆数据和车辆车牌数据采用监督学习的方式进行智能识别训练。

附图说明

图1是本车辆车牌检测定位方法的流程示意图；

图2是YOLOv5网络结构示意图。

具体实施方式

本发明使用多个尺度的YOLOv5模型分别对车辆和车牌进行检测，大幅提高检测的效率。传统的检测算法直接通过整幅图片定位车牌，由于车牌较小，在处理完图片后车牌信息损失。本发明通过模型级联的方式，通过检测出的车辆ROI对原始图像进行裁剪和放大，不会造成车牌信息损失，提高了车牌检测的精度和召回率，使模型具备更强的鲁棒性。

下面结合附图与具体实施方式对本发明进一步说明。