Abstract
Deep learning-based forest fire vision monitoring methods have developed rapidly and are becoming mainstream. The existing methods, however, are based on enormous amounts of data, and have issues with weak feature extraction, poor small target recognition and many missed and false detections in complex forest scenes. In order to solve these problems, we proposed a multi-task learning-based forest fire detection model (MTL-FFDet), which contains three tasks (the detection task, the segmentation task and the classification task) and shares the feature extraction module. In addition, to improve detection accuracy and decrease missed and false detections, we proposed the joint multi-task non-maximum suppression (NMS) processing algorithm that fully utilizes the advantages of each task. Furthermore, considering the objective fact that divided flame targets in an image are still flame targets, our proposed data augmentation strategy of a diagonal swap of random origin is a good remedy for the poor detection effect caused by small fire targets. Experiments showed that our model outperforms YOLOv5-s in terms of mAP (mean average precision) by 3.2%, APS (average precision for small objects) by 4.8%, ARS (average recall for small objects) by 4.0%, and other metrics by 1% to 2%. Finally, the visualization analysis showed that our multi-task model can focus on the target region better than the single-task model during feature extraction, with superior extraction ability.
Funder
National Key R&D Program of China
Reference60 articles.
1. A Survey of Machine Learning Algorithms Based Forest Fires Prediction and Detection Systems
2. A Review on Early Forest Fire Detection Systems Using Optical Remote Sensing
3. Early detection of forest fire based on unmaned aerial vehicle platform;Yang;Proceedings of the 2019 IEEE International Conference on Signal, Information and Data Processing (ICSIDP),2019
4. Real-time forest fire detection with wireless sensor networks;Yu;Proceedings of the 2005 International Conference on Wireless Communications, Networking and Mobile Computing,2005
5. A Wireless Sensor Network Deployment for Rural and Forest Fire Detection and Verification
Cited by
12 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献