交通信号灯识别的自主式学习系统

自动驾驶中一个核心模块的设计。其中算法层面涉及到对交通信号灯的检测与识别，前期对该模块进行了简单算法的设计，主体的识别框架如下图，主要思想是基于滑动窗思想和机器学习模型。当然为了进一步提升算法的效率，采用了基于信号灯颜色的特殊先验知识进行selective search，过滤掉大部分区域实现算法的加速。

上述框架虽然能达到一定的识别率，但是依然面临两个严峻的挑战：

• 对视频处理的实时性较差。上述框架对ROI的提取算法涉及到对整个图像的遍历；并且对视频的每一帧进行独立处理，没有考虑帧与帧之间的相关性。
• 处理不同场景的迁移性较差。上述框架主要基于SVM分类器，而这种传统的机器学习模型的性能主要依赖于训练数据和提取的人工特征。目前的训练数据有限，不能完全表征整个真实数据的分布，导致在其它场景漏检或误检较多；当前的特征主要基于颜色直方图特征，该特征对图像数据较为敏感。比如采用不同的成像传感器或当前采集图像出现极端天气，提取的颜色直方图特征都会给后续的分类器带来误判。

针对上述挑战，对现有框架进行了改进，设计了交通信号灯识别的自主式学习系统来应对实时性和迁移性差的问题。首先我们考虑交通场景中视频的特点，当信号灯出现在当前视野中时，在接下来的一段采集时间内都会出现该信号灯；并且帧与帧之间的信号灯位置不会出现很大的偏移（主要取决于采集车和CCD）且呈现一定的运动方向，如下图。那么我们完全可以利用这种特性，在下一帧的检测中取当前信号灯位置处一定的窗口范围，使用mean shift算法在特征空间中搜索所谓的模点（局部极值点）找到信号灯位置；甚至可以在该窗口检测范围内，使用颜色分割，根据颜色判定信号灯的位置和颜色。注意上述策略只适用于信