[发明专利]烟火识别方法、装置、设备及可读存储介质

权利要求书

1.一种烟火识别方法，其特征在于，包括：

获取第一图像中的待识别的目标对象对应的第一置信度，所述第一置信度用于表示在所述第一图像中所述目标对象为烟火的概率；

在所述第一置信度大于第一预设值的情况下，确定所述第一图像中的目标区域，所述目标区域为包括所述目标对象的图像区域；

对所述目标区域进行放大拍摄，获得K帧第二图像，每一帧所述第二图像包括所述目标区域，K为正整数；

按照至少两种图像识别算法对所述第二图像进行烟火识别，获得所述目标对象对应每一图像识别算法的第二置信度，所述第二置信度用于表示在所述第二图像中所述目标对象为烟火的概率；

根据所述第一置信度和所述第二置信度确定第三置信度，所述第三置信度用于表示所述目标对象为烟火的概率。

2.根据权利要求1所述的方法，其特征在于，所述对所述目标区域进行放大拍摄，获得K帧第二图像之前，还包括：

根据所述目标区域在所述第一图像中的位置信息和大小信息确定拍摄的角度和放大倍数。

3.根据权利要求1所述的方法，其特征在于，所述至少两种图像识别算法包括第一图像识别算法和第二图像识别算法；

按照至少两种图像识别算法对所述第二图像进行烟火识别，获得所述目标对象对应每一图像识别算法的第二置信度包括：

按照所述第一图像识别算法对所述第二图像进行烟火识别，获得第一子置信度；

按照所述第二图像识别算法对所述第二图像进行烟火识别，获得第二子置信度；

其中，所述第二置信度包括所述第一子置信度和所述第二子置信度，所述第一图像识别算法为混合高斯背景建模、形态学滤波处理和canny边缘检测算法的算法，所述第二图像识别算法为YOLOv5算法。

4.根据权利要求3所述的方法，其特征在于，所述按照所述第一图像识别算法对所述第二图像进行烟火识别，获得所述第一子置信度包括：

基于所述K帧第二图像的前M帧图像进行混合高斯背景建模，获得前景图像，所述前景图像用于表示所述K帧第二图像中的目标对象，K和M为正整数；

对所述前景图像进行形态学滤波处理，获得第一目标图像；

对所述第一目标图像进行canny边缘检测算法，获得第二目标图像；

基于所述第二目标图像，获取所述第二图像的第一子置信度；

其中，所述第一目标图像为所述第二图像的一部分。

5.根据权利要求3所述的方法，其特征在于，所述按照所述第二图像识别算法对所述第二图像进行烟火识别，获得第二子置信度包括：

利用YOLOv5算法对所述第二图像进行13*13和26*26两种尺度的检测。

6.根据权利要求4所述的方法，其特征在于，所述基于所述第二目标图像的边缘信息，获取所述第二图像中第二目标图像的第一子置信度包括：

将所述K帧第二图像中像素块面积大于第二预设值的每一帧第二图像的像素块面积合并；

获取所述K帧第二图像中每一帧所述第二图像的增长率及所述增长率分布；

在所述增长率大于第三预设值且满足正态分布时，基于所述增长率和所述K帧第二图像合并后的像素块面积，获取第一子置信度。

7.根据权利要求3所述的方法，其特征在于，所述根据所述第一置信度和所述第二置信度确定第三置信度包括：

根据如下公式确定第三置信度：

式中P_end为第三置信度，W1为第一置信度，W2为第二子置信度，P_areas为第一子置信度，ω根据实际应用场景进行设置。