[发明专利]安全带检测方法、系统、电子设备及可读存储介质

权利要求书

1.一种安全带检测方法，其特征在于，包括：

获取待识别图像，其中，所述待识别图像包括可见光待识别图像和红外待识别图像中的至少一种；

若所述待识别图像包括可见光待识别图像和红外待识别图像，则对所述待识别图像进行图像识别，得到识别结果和识别置信度，其中，所述识别结果包括可见光识别后图像和红外识别后图像；

根据所述识别置信度与预设阈值之间的比较结果从所述识别结果中确定待检测图像，并对所述待检测图像进行安全带检测，得到检测结果。

2.根据权利要求1所述的方法，其特征在于，根据所述识别置信度与预设阈值之间的比较结果从所述识别结果中确定待检测图像，包括：

预先设置第一条件包括所述识别置信度大于或等于所述第一置信度阈值，第二条件包括所述识别置信度大于或等于所述第二置信度阈值，其中，所述第一置信度阈值大于所述第二置信度阈值；

若同时满足所述第一条件和所述第二条件，则将所述可见光识别后图像确定为待检测图像；

若满足所述第二条件且不满足第一条件，则将所述可见光识别后图像和所述红外识别后图像确定为待检测图像；

若同时不满足所述第一条件和所述第二条件，则将所述红外识别后图像确定为待检测图像。

3.根据权利要求2所述的方法，其特征在于，对所述待检测图像进行安全带检测，得到检测结果，包括：

若所述待检测图像包括所述可见光识别后图像或所述红外识别后图像，则将所述待检测图像输入安全带检测模型，得到模型输出结果，并根据所述模型输出结果确定检测结果；

若所述待检测图像包括所述可见光识别后图像和所述红外识别后图像，则将所述待检测图像输入安全带检测模型，分别得到可见光输出结果和红外输出结果，根据所述可见光输出结果和所述红外输出结果确定综合输出结果，并根据所述综合输出结果确定检测结果。

4.根据权利要求3所述的方法，其特征在于，通过以下公式确定综合输出结果：

Conf＝w₁·r_visible+w₂·r_swir，

其中，Conf为综合输出结果，r_visible为所述可见光输出结果，r_swir为所述红外输出结果，w₁为第一预设权重，w₂为第二预设权重。

5.根据权利要求1至4任一项所述的方法，其特征在于，获取待识别图像，包括：

获取同一目标图像对应的可见光待识别图像和红外待识别图像；

确定所述可见光待识别图像的光照度参数，并将所述光照度参数与预设光照度阈值进行比较，得到比较结果；

若比较结果指示所述光照度参数大于或等于预设光照度阈值，则将所述可见光待识别图像和所述红外待识别图像确定为待识别图像；

若比较结果指示所述光照度参数小于预设光照度阈值，则将所述红外待识别图像确定为待识别图像。

6.根据权利要求5所述的方法，其特征在于，获取待识别图像之后，所述方法还包括：

若所述待识别图像包括可见光待识别图像或红外待识别图像，则对所述待识别图像进行图像识别，得到识别后图像；

将所述识别后图像确定为待检测图像，并对所述待检测图像进行安全带检测，得到检测结果。

7.根据权利要求5所述的方法，其特征在于，确定所述可见光待识别图像的光照度参数，包括：

从所述可见光待识别图像中确定参考区域，并获取所述参考区域的亮度值；

根据所述参考区域的亮度值确定所述可见光场景图像的光照度参数。