The embodiment of the invention discloses a traffic light color tracing method and device, equipment and a storage medium, and the method comprises the steps: obtaining a to-be-processed image frame set corresponding to a to-be-processed video stream, carrying out the color recognition and conversion of each to-be-processed image frame in the to-be-processed image frame set, and generating an auxiliary image frame set corresponding to the to-be-processed image frame set; the lightening pixel in each auxiliary image frame in the auxiliary image frame set is a green pixel or a preset non-green pixel; determining each to-be-processed image frame and the corresponding auxiliary image frame as a to-be-processed image frame group according to the corresponding relationship, respectively inputting each to-be-processed image frame group into a pre-trained traffic light color tracing model, and determining a preliminary color tracing image frame set; and according to the pixel category information in each preliminary coloring image frame and the traffic light position information in the to-be-processed image frame, performing color restoration on each preliminary coloring image frame. Recognition confusion of red and yellow lights in the traffic lights is reduced, and the accuracy of traffic light depicting is improved.
本申请实施例公开了一种交通灯描色方法、装置、设备及存储介质,包括:获取待处理视频流对应的待处理图像帧集合,并将待处理图像帧集合中各待处理图像帧进行颜色识别转换,生成与待处理图像帧集合对应的辅助图像帧集合;辅助图像帧集合中各辅助图像帧中的点亮像素为绿像素或预设非绿像素;根据对应关系将每个待处理图像帧与对应的辅助图像帧确定为一个待处理图像帧组,将各待处理图像帧组分别输入至预训练的交通灯描色模型,确定初步描色图像帧集合;根据各初步描色图像帧中的像素类别信息,以及待处理图像帧中的交通灯位置信息,对各初步描色图像帧进行颜色还原。减少了交通灯中红黄灯识别混淆,提升了交通灯描色的准确率。
Traffic light depicting method, device and equipment and storage medium
交通灯描色方法、装置、设备及存储介质
2023-09-29
Patent
Elektronische Ressource
Chinesisch
IPC: | G06T Bilddatenverarbeitung oder Bilddatenerzeugung allgemein , IMAGE DATA PROCESSING OR GENERATION, IN GENERAL / G06N COMPUTER SYSTEMS BASED ON SPECIFIC COMPUTATIONAL MODELS , Rechnersysteme, basierend auf spezifischen Rechenmodellen / G06V / G08G Anlagen zur Steuerung, Regelung oder Überwachung des Verkehrs , TRAFFIC CONTROL SYSTEMS / H04N PICTORIAL COMMUNICATION, e.g. TELEVISION , Bildübertragung, z.B. Fernsehen |
Traffic light signal control method device equipment and storage medium
Europäisches Patentamt | 2021
|Traffic light signal control method device equipment and storage medium
Europäisches Patentamt | 2023