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

权利要求书

1.一种训练数据获取方法，包括：

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

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

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

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

2.根据权利要求1所述的方法，还包括：

基于所述训练数据对颜色识别模型进行训练，得到训练完成后的颜色识别模型，其中，训练完成后的所述颜色识别模型能够对待识别视频帧所展示的信号指示组件中指示灯的颜色进行识别。

3.根据权利要求1或2所述的方法，其中，所述确定所述信号指示组件中所述指示灯的灰度图像，包括：

确定所述信号指示组件在所述视频帧中所在位置区域；

基于所述位置区域，确定出所述信号指示组件中所述指示灯的灰度图像。

4.根据权利要求1所述的方法，还包括：

将不同所述指示灯的灰度图像进行差分处理，得到差分图像；

确定所述差分图像的特征值；

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

基于所述信号指示组件中所述指示灯的亮度特征，以及所述差分图像的特征值，确定出所述信号指示组件的工作状态。

5.根据权利要求4所述的方法，所述基于所述信号指示组件中所述指示灯的亮度特征，以及所述差分图像的特征值，确定出所述信号指示组件的工作状态，包括：

基于所述信号指示组件中所述指示灯的亮度特征，以及所述差分图像的特征值，确定出所述信号指示灯处于第一目标状态，其中，所述第一目标状态下所有所述指示灯均处于开启状态或者均处于关闭状态，所述第一目标状态满足所述预设条件。

6.根据权利要求4所述的方法，所述基于所述信号指示组件中所述指示灯的亮度特征，以及所述差分图像的特征值，确定出所述信号指示组件的工作状态，包括：

基于所述信号指示组件中所述指示灯的亮度特征，以及所述差分图像的特征值，确定出所述信号指示灯处于第二目标状态，其中，所述第二目标状态下至少两个所述指示灯处于开启状态，所述第二目标状态满足所述预设条件。

7.根据权利要求4或5或6所述的方法，其中，所述确定所述差分图像的特征值，包括：

基于所述差分图像中像素点的像素值与像素阈值的比较结果，对所述差分图像中的像素点进行处理，得到像素点处理后的差分图像；

至少将像素点处理后的所述差分图像中所有像素点的像素值进行归一化处理后，得到针对所述差分图像的特征值。

8.根据权利要求7所述的方法，还包括：

确定所述信号指示组件在所述视频帧中所在位置区域的面积；

其中，所述至少将像素点处理后的所述差分图像中所有像素点的像素值进行归一化处理后，得到针对所述差分图像的特征值，包括：

将像素点处理后的所述差分图像中所有像素点的像素值进行累加处理；

基于累加处理后的总像素值与所述面积的比值，得到针对所述差分图像的特征值。

9.一种训练数据获取装置，包括：

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

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

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

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

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