摘要
快速准确地检测森林火灾中的火点对于减轻火灾损失和开展有效的灭火救援工作至关重要。深度学习技术可以自动学习并提取不同传感器获取的林火特征。无人机搭载不同类型的传感器可以快速获取森林实时图像,但不同传感器获取的图像特征不同,导致深度学习模型在不同图像中的迁移性较差。本研究基于YOLOv5预训练模型,使用286张可见光和红外图像进行迁移学习,以增强模型对不同图像类型的适用性。迁移学习后,模型对可见光图像测试集的检测精度提升了6%,对红外图像测试集的检测精度达到0.952,证明模型对两种传感器图像数据均具有较强的检测能力。相比可见光模型,多模态模型在包含可见光和红外图像的测试集上的mAP50达到0.914,表明模型成功地提高了对红外图像的检测能力,并保留了对可见光图像的良好检测性能。综上,本研究采用少量图像进行迁移学习,取得了较好结果。迁移学习可以使YOLOv5火灾检测模型适配红外图像,在保留可见光图像检测能力的同时,增强其对火情的检测准确性与环境适应性。这为实现基于多源数据的森林火灾检测奠定了基础,为提高森林火灾检测系统的智能化与自适应提供了思路。
Rapid and accurate detection of fire spots in forest fires is crucial for reducing fire losses and conducting effective firefighting and rescue operations.Deep learning techniques can automatically learn and extract forest fire features obtained from different sensors.Unmanned aerial vehicles equipped with different types of sensors can quickly obtain real-time images of forests,but the different image features acquired by different sensors lead to poor transferability of deep learning models in different images.In this study,based on the YOLOv5 pre-trained model for fire detection,we used 286 visible light and infrared images for transfer learning to enhance the model’s applicability to different types of images.After transfer learning,the model achieved a detection accuracy of 0.761 for the visible light image test set and 0.952 for the infrared image test set.This demonstrates that the model has strong detection capabilities for both types of sensor image data.Compared with the original model,the mAP50 of the model after transfer learning on the test set containing visible light and infrared images reached 0.914.This indicates that the model successfully improved its detection ability for infrared images while preserving good detection performance for visible light images.In summary,this study used a small number of images for transfer learning and achieved good results.Transfer learning can adapt the YOLOv5 fire detection model to infrared images,enhancing its detection accuracy and environmental adaptability for fire situations while retaining the detection capabilities for visible light images.This lays the foundation for realizing multi-source data-based forest fire detection and provides ideas for improving the intelligence and adaptability of forest fire detection systems.
作者
谈宜院
刘昊辰
陈礼波
沈毓芬
刘学辉
佃袁勇
Tan Yiyuan;Liu Haochen;Chen Libo;Shen Yufen;Liu Xuehui;Dian Yuanyong(Forest Fire Prevention Center,Xianning City,Hubei Province,Xianning 437000;College of Horticulture and Forestry,Huazhong Agricultural University,Wuhan 430070;Hubei Forestry Information Engineering Technology Research Center,Wuhan 430070)
出处
《湖北林业科技》
2023年第6期33-37,69,共6页
Hubei Forestry Science and Technology
