[发明专利]一种适用于无人机平台的火灾实时检测方法及系统

说明书

本发明涉及一种适用于无人机平台的火灾实时检测方法及系统。获取无人机火灾图像数据集；对无人机火灾图像数据集进行预处理，得到预处理后的无人机火灾图像数据集；建立改进后的YOLOv5网络模型；对预处理后的无人机火灾图像数据集进行训练，得到无人机火灾检测模型权重；根据无人机火灾检测模型权重，将预处理后的无人机火灾图像数据集输入至改进后的YOLOv5网络模型，得到火灾实时检测模型；对火灾实时检测模型的性能进行验证、测试和评估。本发明能够快速准确地针对无人机图像特点进行火灾实时检测。

技术领域

本发明涉及目标检测领域，特别是涉及一种适用于无人机平台的火灾实时检测方法及系统。

背景技术

火灾是世界性的自然灾害之一，也是日常生活中主要灾害，其发生发展不仅对自然环境有着严重影响，还威胁着人们的生命与财产安全，因此及时准确地发现火灾并预警具有十分重要的研究和现实意义。近年来由于自然和人为原因引发的火灾已造成极大的人员伤亡和生态破坏。

常规无人机火灾检测通过无人机搭载温度、烟雾传感器采集相关数据进行相关处理后实现火灾检测，但是检测准确性和实时性得不到保证。随着计算机技术和人工智能技术的不断发展，深度学习技术已经在图像识别和检测方面展示了良好性能，被广泛应用到火灾检测和预警。现有技术中，申请号为202110358662.6公开了一种基于无人机的森林火灾检测系统，其无人机平台搭载红外热成像仪、空气采集装置等，通过热成像图像判断巡查区域是否存在火灾隐患，也可通过分析空气样本中的可燃气体、烟雾浓度判断空气采集点是否存在火灾隐患，但成本较高，火灾信息处理速度较慢，实时性较差。申请号为202211629322.3公开了一种基于无人机实时视频流的火灾检测方法及系统，火灾数据为常规火灾图像，以Darknet为训练框架，YOLOv4-Tiny为基础模型，通过改进原模型激活函数、增加注意力机制来提高模型性能，通过模型剪枝降低模型参数量以期可适用于无人机平台，但其训练数据集为常规火灾图像，改进方法并未针对无人机图像特点，不具有在无人机平台部署的条件。

综上所述，现有技术中亟需一种适用于无人机平台的火灾实时检测方法及系统。

发明内容

本发明的目的是提供一种适用于无人机平台的火灾实时检测方法及系统，能够快速准确地针对无人机图像特点进行火灾实时检测。

为实现上述目的，本发明提供了如下方案：

一种适用于无人机平台的火灾实时检测方法包括：

获取无人机火灾图像数据集；

对所述无人机火灾图像数据集进行预处理，得到预处理后的无人机火灾图像数据集；

建立改进后的YOLOv5网络模型；

对所述预处理后的无人机火灾图像数据集进行训练，得到无人机火灾检测模型权重；

根据所述无人机火灾检测模型权重，将所述预处理后的无人机火灾图像数据集输入至所述改进后的YOLOv5网络模型，得到火灾实时检测模型；

对所述火灾实时检测模型的性能进行验证、测试和评估。

可选地，还包括：

将所述火灾实时检测模型部署至无人机平台。

可选地，所述对所述无人机火灾图像数据集进行预处理，得到预处理后的无人机火灾图像数据集，具体包括：

对所述无人机火灾图像数据集进行筛选，剔除不符合要求的图像，得到筛选数据集；

对所述筛选数据集通过图像旋转、裁剪、镜像、加噪的方法扩充数据集，得到泛化后的数据集；

采用Labelme软件对所述泛化后的数据集所有图像进行人工标注。

可选地，所述建立改进后的YOLOv5网络模型，具体包括：