基于深度学习指示灯检测及颜色识别泛化能力提升算法

摘要

本发明公开了基于深度学习指示灯检测及颜色识别泛化能力提升算法，该算法包括以下步骤：收集需要识别的指示灯照片，并通过预先配置的标注方法对照片中指示灯的位置进行标注，保存标注结果；搭建深度学习神经网络，对已标注的照片进行特征提取得到标注数据，且对损失函数进行构建，并通过标注数据及损失函数对搭建的网络进行训练及方向推导，保存训练的模型；利用训练好的模型对需要识别的照片进行识别和检测，并将检测得到的色域与预先设定的颜色区域进行比较筛选和处理，得到指示灯颜色识别结果。有益效果：本发明识别速度更快，识别更为准确，对照片拍摄设备、指示灯、拍摄条件要求低，降低了算法算力要求，节省算法部署调试的时间。