一种交通信号灯识别方法和装置

摘要

本发明公开了一种交通信号灯识别方法和装置，属于交通安全领域。所述方法包括：采集原始图像；对原始图像在RGB空间的三个通道值进行阈值分割，得到黑色区域二值图像；根据灯板的特征，在黑色区域二值图像中定位出灯板区域；将原始图像中灯板区域的图像从RGB空间转换到YCbCr空间，分割出红色和黄色区域二值图像以及绿色区域二值图像；根据交通信号灯的特征，在红色和黄色区域二值图像以及绿色区域二值图像上确定交通信号灯候选区域；将原始图像中交通信号灯候选区域的图像灰度化和归一化，对灰度化和归一化后的图像进行Gabor小波变换，得到Gabor小波图像；对Gabor小波图像的幅值进行采样，得到特征向量；比较特征向量与训练样本的相似度，确定交通信号灯的类别。

主权项

一种交通信号灯识别方法，其特征在于，所述方法包括：采集原始图像；对所述原始图像在红绿蓝RGB空间的三个通道值进行阈值分割，得到黑色区域二值图像；根据灯板的特征，在所述黑色区域二值图像中定位出灯板区域，所述灯板的特征包括所述灯板的宽高比以及所述灯板在所述原始图像中的位置；将所述原始图像中所述灯板区域的图像从RGB空间转换到YCbCr空间，并分别分割出红色和黄色区域二值图像以及绿色区域二值图像；根据所述交通信号灯的特征，在所述红色和黄色区域二值图像以及绿色区域二值图像上确定交通信号灯候选区域，所述交通信号灯的特征包括所述交通信号灯的外接矩形的宽高比以及所述交通信号灯与所述灯板区域的相对位置；将所述原始图像中所述交通信号灯候选区域的图像灰度化和归一化，并对灰度化和归一化之后的所述交通信号灯候选区域的图像进行Gabor小波变换，得到Gabor小波图像；对所述Gabor小波图像的幅值进行采样，得到特征向量；比较所述特征向量与训练样本库中训练样本的相似度，确定所述交通信号灯的类别，所述训练样本库包括各个类别的交通信号灯的训练样本。