[发明专利]一种交通限速标志检测方法、装置、设备及存储介质

说明书

本申请公开了一种交通限速标志检测方法、装置、设备及存储介质。该方法包括：获取待检测图像，并将待检测图像转换为HSV图像；根据HSV图像中各像素点的颜色参数，分别利用红色对应的第一颜色参数阈值和蓝色对应的第二颜色参数阈值对HSV图像进行二值化处理，以得到对应的红色阈值二值图像和蓝色阈值二值图像；根据红色阈值二值化图像和蓝色阈值二值化图像的像素值分布，通过分割从待检测图像中提取出目标区域图像，并通过拼接目标区域图像得到目标拼接图像；利用预先经过历史目标拼接图像训练得到的目标检测模型对目标拼接图像进行交通限速标志检测，并将检测结果框还原至待检测图像以得到检测后图像。提高了交通限速标志检测的准确性和速度。

技术领域

本发明涉及图像处理技术领域，特别涉及一种交通限速标志检测方法、装置、设备及存储介质。

背景技术

当前，限速标志检测是智能交通的重要前提和基础，由于背景的复杂性，且限速标志目标过小，对其仅采用目标识别的方法检测十分困难，召回率和准确率均不高。再加上车辆在道路上行驶时，速度比较快，增加了检测难度。

现有技术中，限速标志检测主要包括以下几种：

(1)直接对全图进行分块目标检测。即直接对全图进行分块目标检测的方法，但会使得一张图片至少需要检测两次，检测的速度明显下降。

(2)利用圆形辅助提取限速标志区域再对该区域进行检测。基于圆形检测结合目标识别算法进行交通标志检测，但其进行圆形检测涉及到大量的计算，预处理会耗费大量的时间，也会相应的拖慢单帧检测的时间。

(3)使用颜色辅助提取限速标志区域再对该区域进行检测。具体利用HSV颜色空间和RGB颜色空间对不同颜色的交通标志进行粗检测，然后利用模板匹配的方法对交通标志进行检测，但该方法粗检测的效果较差，且模版匹配降低了检测速度。

因此如何及时准确地检测出道路旁的限速标志是目前亟需解决的问题。

发明内容

有鉴于此，本发明的目的在于提供一种交通限速标志检测方法、装置、设备及介质，能够提高交通限速标志检测的准确度和速度。其具体方案如下：

第一方面，本申请公开了一种交通限速标志检测方法，包括：

获取待检测图像，并将所述待检测图像转换为HSV图像；

根据所述HSV图像中各像素点的颜色参数，分别利用红色对应的第一颜色参数阈值和蓝色对应的第二颜色参数阈值对所述HSV图像进行二值化处理，以得到对应的红色阈值二值图像和蓝色阈值二值图像；其中，所述颜色参数包括色调、饱和度和明度；

根据所述红色阈值二值化图像和蓝色阈值二值化图像的像素值分布，通过分割从所述待检测图像中提取出目标区域图像，并通过拼接所述目标区域图像得到目标拼接图像；

利用预先经过历史目标拼接图像训练得到的目标检测模型对所述目标拼接图像进行交通限速标志检测，并将检测结果框还原至所述待检测图像以得到检测后图像。

可选的，所述根据所述红色阈值二值化图像和蓝色阈值二值化图像的像素值分布，通过分割从所述待检测图像中提取出目标区域图像，并通过拼接所述目标区域图像得到目标拼接图像，包括：

根据所述红色阈值二值化图像的像素值分布，通过按行分割从所述待检测图像中提取出目标行区域图像，并通过拼接所述目标行区域图像得到行片段原图；

根据所述行片段原图在红色阈值和蓝色阈值下的二值化片段图像，通过按列分割从所述行片段原图中提取出目标列区域图像，并通过拼接所述目标列区域图像得到列片段原图；

基于所述列片段原图得到所述目标拼接图像。