[发明专利]一种烟火检测方法、装置、设备及计算机可读存储介质

说明书

本申请涉及一种烟火检测方法、装置、设备及计算机可读存储介质，属于图像识别技术领域，包括获取待检测图像；将所述待检测图像输入至目标检测模型，并基于所述目标检测模型输出待检测图像的预测框信息；其中，所述目标检测模型包括普通卷积模块、可变卷积模块、特征融合模块以及后处理模块，所述预测框信息包括目标类别、置信度以及所述预测框在所述待检测图像内的位置信息；根据所述预测框信息对所述待检测图像中的烟火区域进行标记，得到已标记图像；将可变卷积模块利用到目标检测模型中，便于目标检测模型训练出与烟火实际形状相匹配的卷积核形态，对烟火的检测更加准确；本申请具有便于对烟火目标进行检测的效果。

技术领域

本发明涉及图像识别技术领域，尤其是涉及一种烟火检测方法、装置、设备及计算机可读存储介质。

背景技术

烟火检测主要用来检测烟火、预防火灾，便于人们及时得知火灾信息，以便及时对火灾进行控制、处理。目前，基于视频的烟火检测方法，大多先采用帧间差分法、背景建模法等检测出候选运动区域，再对候选运动区域提取颜色、纹理等特征，用分类器判定候选运动区域是否是烟火，当发生烟火的距离比较远时，由于视频上烟火的像素比较小无法进行判断，从而引起漏报的现象。目标检测，是一种基于目标几何和统计特征的图像分割，其准确性和实时性是整个系统的一项重要能力，尤其是在复杂场景中，需要对多个目标进行实时处理时，目标自动提取和识别就显得特别重要；若将目标检测的方法应用到烟火检测中，针对复杂场景，利用目标检测往往有较好的检测效果。

在实现本申请过程中，发明人发现至少存在如下问题：目标检测往往依赖基于特征提取的边界框，采用目标检测对烟火进行检测时，不能很好的适应烟火的实际形态，导致利用目标检测的方法对烟火等不规则目标的检测效果不够理想。

发明内容

为了便于对烟火目标进行检测，本申请提供了一种烟火检测方法、装置、设备及计算机可读存储介质。

第一方面，本申请提供的一种烟火检测方法采用如下的技术方案：

一种烟火检测方法，包括，

获取待检测图像；

将所述待检测图像输入至目标检测模型，生成待检测图像的预测框信息；其中，所述目标检测模型包括普通卷积模块、可变卷积模块、特征融合模块以及后处理模块；

根据所述预测框信息对所述待检测图像中的烟火区域进行标记，得到已标记图像。

通过采用上述技术方案，将获取到的待检测图像输入至目标检测模型中，目标检测模型输出预测框信息，由于将可变卷积模块应用至标检测模型中，在对目标检测模型进行训练时，能够训练出动态的卷积核，使得卷积核的形态不局限在矩形，即目标检测模型的检测范围能够更加贴合烟火的实际形状，再根据预测框信息对待检测图像中的烟火区域进行标记，得到已标记图像，从而实现了便于对烟火进行检测的效果。

可选的，所述根据所述预测框信息对所述待检测图像中的烟火区域进行标记的步骤，具体包括：

根据预设切分重叠度将待检测图像切分为多张子图；其中，所述预设切分重叠度是切分后相邻子图的重叠度；

将多张所述子图分别输入至所述目标检测模型中，生成多个子图预测框信息；

基于所述待检测图像的预测框信息，获得所述待检测图像的目标类别和置信度；

基于多个子图预测框信息，分别获得所述多个子图对应的目标类别和置信度；

判断任一子图的目标类别是否与所述待检测图像的目标类别一致，若是，判断所述目标类别一致的子图的置信度和所述待检测图像的置信度是否均大于预设置信度，若是，则根据所述待检测图像的预测框信息对所述待检测图像中的烟火区域进行标记。