[发明专利]红绿灯识别质量的评价方法、装置和电子设备

权利要求书

1.一种红绿灯识别质量的评价方法，其特征在于，包括：

根据红绿灯识别算法识别到的交通灯的颜色和标注的交通灯的颜色确定所述红绿灯识别算法识别的颜色是否正确；

若正确，则计算所述红绿灯识别算法识别到的交通灯的包围框与标注的交通灯的包围框的重叠率，并将所述重叠率作为第一得分；

统计所述红绿灯识别算法识别到的正确颜色的像素的像素数量，并根据所述像素数量和对应的交通灯的像素总数量计算第二得分；

根据所述第一得分和所述第二得分确定所述红绿灯识别质量的得分。

2.根据权利要求1所述的方法，其特征在于，所述方法还包括：

若所述红绿灯识别算法识别的颜色不正确，则确定所述红绿灯识别质量的得分为预设得分。

3.根据权利要求1所述的方法，其特征在于，根据红绿灯识别算法识别到的交通灯的颜色和标注的交通灯的颜色确定所述红绿灯识别算法识别的颜色是否正确，包括：

获取测试数据集，其中，所述测试数据集中包括：交通灯图像、交通灯图像对应的标注的交通灯的颜色；

采用所述红绿灯识别算法对所述交通灯图像进行红绿灯识别，得到交通灯图像对应的交通灯的颜色；

根据所述交通灯图像对应的交通灯的颜色和所述交通灯图像对应的标注的交通灯的颜色确定所述红绿灯识别算法识别的颜色是否正确。

4.根据权利要求3所述的方法，其特征在于，计算所述红绿灯识别算法识别到的交通灯的包围框与标注的交通灯的包围框的重叠率，包括：

获取所述红绿灯识别算法对所述交通灯图像进行识别得到的交通灯的包围框；

计算所述识别得到的交通灯的包围框与对应的标注的交通灯的包围框的交并比；

将所述交并比作为所述重叠率。

5.根据权利要求1所述的方法，其特征在于，根据所述像素数量和对应的交通灯的像素总数量计算第二得分，包括：

计算所述像素数量与所述像素总数量之间的比值；

将所述比值作为所述第二得分。

6.根据权利要求1所述的方法，其特征在于，统计所述红绿灯识别算法识别到的正确颜色的像素的像素数量，包括：

获取待识别像素的像素位置，其中，所述像素位置为根据标注的交通灯图像确定的；

根据所述像素位置确定待检测图像中对应的目标像素；

根据所述目标像素的HSV值统计所述目标像素中为所述红绿灯识别算法识别到的交通灯的颜色的像素的像素数量。

7.根据权利要求1所述的方法，其特征在于，所述方法还包括：

若所述红绿灯识别质量的得分小于预设阈值，则对所述红绿灯识别算法进行优化训练。

8.一种红绿灯识别质量的评价装置，其特征在于，包括：

第一确定单元，用于根据红绿灯识别算法识别到的交通灯的颜色和标注的交通灯的颜色确定所述红绿灯识别算法识别的颜色是否正确；

计算单元，用于若正确，则计算所述红绿灯识别算法识别到的交通灯的包围框与标注的交通灯的包围框的重叠率，并将所述重叠率作为第一得分；

统计并计算单元，用于统计所述红绿灯识别算法识别到的正确颜色的像素的像素数量，并根据所述像素数量和对应的交通灯的像素总数量计算第二得分；

第二确定单元，用于根据所述第一得分和所述第二得分确定所述红绿灯识别质量的得分。

9.一种电子设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时实现上述权利要求1至7中任一项所述的方法的步骤。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质存储有机器可运行指令，所述机器可运行指令在被处理器调用和运行时，所述机器可运行指令促使所述处理器运行上述权利要求1至7中任一项所述的方法。