[发明专利]训练数据获取方法、装置、电子设备及存储介质

说明书

本申请公开了训练数据获取方法、装置、电子设备及存储介质，涉及大数据、智能交通领域。具体实现方案为：获取视频帧，其中，所述视频帧为从视频数据中选取出的，并展示有信号指示组件，所述信号指示组件中包含有至少两个指示灯；确定所述信号指示组件中所述指示灯的灰度图像，并基于所述灰度图像确定出所述指示灯的亮度特征；至少基于所述信号指示组件中所述指示灯的亮度特征确定出所述信号指示组件的工作状态；从所述视频数据中选取出所述信号指示组件的工作状态满足预设条件的视频帧作为训练数据。如此，实现自动化获取训练数据的目标，同时，丰富了训练数据所涵盖的范围。

技术领域

本申请涉及数据处理领域，尤其涉及大数据、智能交通领域。

背景技术

在颜色识别模型(比如，用于对红绿灯的灯色进行识别的模型)的训练过程中，需要先采集红绿灯数据，然后将采集的红绿灯数据作为训练数据来对颜色识别模型进行训练。而实际应用中，训练数据所涵盖的范围在很大程度上决定了颜色识别模型的准确率。因此，如何丰富训练数据所涵盖的范围成为提升颜色识别模型的识别效果的主要因素之一。

发明内容

本申请提供了一种训练数据获取方法、装置、电子设备及存储介质。

根据本申请的一方面，提供了一种训练数据获取方法，包括：

获取视频帧，其中，所述视频帧为从视频数据中选取出的，并展示有信号指示组件，所述信号指示组件中包含有至少两个指示灯；

确定所述信号指示组件中所述指示灯的灰度图像，并基于所述灰度图像确定出所述指示灯的亮度特征；

至少基于所述信号指示组件中所述指示灯的亮度特征确定出所述信号指示组件的工作状态；

从所述视频数据中选取出所述信号指示组件的工作状态满足预设条件的视频帧以作为训练数据。

根据本申请的另一方面，提供了一种训练数据获取装置，包括：

视频帧获取单元，用于获取视频帧，其中，所述视频帧为从视频数据中选取出的，并展示有信号指示组件，所述信号指示组件中包含有至少两个指示灯；

灰度图像处理单元，用于确定所述信号指示组件中所述指示灯的灰度图像；

亮度特征处理单元，用于基于所述灰度图像确定出所述指示灯的亮度特征；

工作状态确定单元，用于至少基于所述信号指示组件中所述指示灯的亮度特征确定出所述信号指示组件的工作状态；

训练数据确定单元，用于从所述视频数据中选取出所述信号指示组件的工作状态满足预设条件的视频帧以作为训练数据。

根据本申请的再一方面，提供了一种电子设备，包括：

至少一个处理器；以及

与所述至少一个处理器通信连接的存储器；其中，

所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行以上所述的方法。

根据本申请的再一方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，其特征在于，所述计算机指令用于使所述计算机执行以上所述的方法。

根据本申请的再一方面，提供了一种路侧设备，所述设备包括：

一个或多个处理器；以及

存储装置，用于存储一个或多个程序，所述一个或多个程序被所述一个或多个处理器执行，以使所述一个或多个处理器实现以上所述的方法。

根据本申请的技术实现了自动化获取训练数据的目标，同时，丰富了训练数据所涵盖的范围。