LEF-YOLO: a lightweight method for intelligent detection of four extreme wildfires based on the YOLO framework
Jianwei Li A * , Huan Tang A , Xingdong Li B , Hongqiang Dou C and Ru Li DA
B
C
D
Abstract
Extreme wildfires pose a serious threat to forest vegetation and human life because they spread more rapidly and are more intense than conventional wildfires. Detecting extreme wildfires is challenging due to their visual similarities to traditional fires, and existing models primarily detect the presence or absence of fires without focusing on distinguishing extreme wildfires and providing warnings.
To test a system for real time detection of four extreme wildfires.
We proposed a novel lightweight model, called LEF-YOLO, based on the YOLOv5 framework. To make the model lightweight, we introduce the bottleneck structure of MobileNetv3 and use depthwise separable convolution instead of conventional convolution. To improve the model’s detection accuracy, we apply a multiscale feature fusion strategy and use a Coordinate Attention and Spatial Pyramid Pooling-Fast block to enhance feature extraction.
The LEF-YOLO model outperformed the comparison model on the extreme wildfire dataset we constructed, with our model having excellent performance of 2.7 GFLOPs, 61 FPS and 87.9% mAP.
The detection speed and accuracy of LEF-YOLO can be utilised for the real-time detection of four extreme wildfires in forest fire scenes.
The system can facilitate fire control decision-making and foster the intersection between fire science and computer science.
Keywords: convolutional neural networks, deep learning, extreme wildfire, fire safety, lightweight, multiscale feature fusion, object detection, YOLO (LEF-YOLO).
References
Almeida JS, Huang C, Nogueira FG, Bhatia S, De Albuquerque VHC (2022) EdgeFireSmoke: A Novel Lightweight CNN Model for Real-Time Video Fire-Smoke Detection. IEEE Transactions on Industrial Informatics 18(11), 7889-7898.
| Crossref | Google Scholar |
Azim MR, Keskin M, Do N, Gül M (2022) Automated classification of fuel types using roadside images via deep learning. International Journal of Wildland Fire 31(10), 982-987.
| Crossref | Google Scholar |
Bao W, Yang X, Liang D, Hu G, Yang X (2021) Lightweight convolutional neural network model for field wheat ear disease identification. Computers and Electronics in Agriculture 189(4), 106367.
| Crossref | Google Scholar |
Barmpoutis P, Dimitropoulos K, Kaza K, Grammalidis N (2019) Fire Detection from Images Using Faster R-CNN and Multidimensional Texture Analysis. In ‘ICASSP 2019 - 2019 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP)’, 12–17 May 2019, Brighton, UK. pp. 8301–8305. 10.1109/ICASSP.2019.8682647
Castro RF, Morgan P, Fernandes P, Hoffman C (2021) Extreme Fires. In ‘Fire Science’. pp. 175–257. (Springer: Cham) 10.1007/978‐3‐030‐69815‐7_8
Chen A (2022) Evaluating the relationships between wildfires and drought using machine learning. International Journal of Wildland Fire 31(3), 230-239.
| Crossref | Google Scholar |
Chen XW, Hopkins B, Wang H, O’Neill L, Afghah F, Razi A, Fule P, Coen J, Rowell E, Watts A (2022) Wildland Fire Detection and Monitoring Using a Drone-Collected RGB/IR Image Dataset. IEEE Access 10, 121301-121317.
| Crossref | Google Scholar |
Cui MD, Lou YY, Ge YL, Wang KQ (2023) LES-YOLO: A lightweight pinecone detection algorithm based on improved YOLOv4-Tiny network. Computers and Electronics in Agriculture 205, 107613.
| Crossref | Google Scholar |
Dai Y, Liu W, Wang H, Xie W, Long K (2022) YOLO-Former: Marrying YOLO and Transformer for Foreign Object Detection. IEEE Transactions on Instrumentation and Measurement 71, 1-14.
| Crossref | Google Scholar |
Dang FY, Chen D, Lu YZ, Li ZJ (2023) YOLOWeeds: A novel benchmark of YOLO object detectors for multi-class weed detection in cotton production systems. Computers and Electronics in Agriculture 205, 107655.
| Crossref | Google Scholar |
Department of Forestry and Fire Management (2013) Yarnell Hill Fire Report Now Available. Available at https://dffm.az.gov/yarnell-hill-report-available [verified 23 September 2013]
Diao ZH, Yan JN, He ZD, Zhao SN, Guo PL (2022) Corn seedling recognition algorithm based on hyperspectral image and lightweight-3D-CNN. Computers and Electronics in Agriculture 201, 107343.
| Crossref | Google Scholar |
Gómez-Vázquez I, Fernandes PM, Arias-Rodil M, Barrio-Anta M, Castedo-Dorado F (2014) Using density management diagrams to assess crown fire potential in Pinus pinaster Ait. stands. Annals of Forest Science 71(4), 473-484.
| Crossref | Google Scholar |
Gonzalez-Huitron V, León-Borges JA, Rodriguez-Mata AE, Amabilis-Sosa LE, Ramírez-Pereda B, Rodriguez H (2021) Disease detection in tomato leaves via CNN with lightweight architectures implemented in Raspberry Pi 4. Computers and Electronics in Agriculture 181, 105951.
| Crossref | Google Scholar |
Hantson S, Andela N, Goulden ML, Randerson JT (2022) Human-ignited fires result in more extreme fire behavior and ecosystem impacts. Nature Communications 13(1), 2717.
| Crossref | Google Scholar | PubMed |
Hou Q, Zhou D, Feng J (2021) Coordinate Attention for Efficient Mobile Network Design. In ‘2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)’, Nashville, TN, USA, 20–25 June 2021. pp. 13708–13717. 10.1109/CVPR46437.2021.01350
Howard A, Sandler M, Chu G, Chen LC, Chen B, Tan MX, Wang WJ, Zhu YK, Pang RM, Vasudevan V, Le QV, Adam H (2019) Searching for MobileNetV3. In ‘2019 IEEE/CVF International Conference on Computer Vision (ICCV)’, Seoul, Korea, 27 October–2 November 2019. pp. 1314–1324. 10.1109/ICCV.2019.00140
Hsu WY, Lin WY (2021) Ratio-and-Scale-Aware YOLO for Pedestrian Detection. IEEE Transactions on Image Processing 30, 934-947.
| Crossref | Google Scholar | PubMed |
Jocher G (2020) yolov5. Code repository. Available at https://github.com/ultralytics/yolov5
Lei J, Deng WY, Mao SH, Tao Y, Wu HG, Xie CG (2023) Flame geometric characteristics of large-scale pool fires under controlled wind conditions. Proceedings of the Combustion Institute 39(3), 4021-4029.
| Crossref | Google Scholar |
Li JW, Li XW, Chen CC, Zheng HR, Liu NY (2018) Three-dimensional dynamic simulation system for forest surface fire spreading prediction. International Journal of Pattern Recognition and Artificial Intelligence 32(8), 1850026.
| Crossref | Google Scholar |
Lin TY, Dollar P, Girshick R, He KM, Hariharan B, Belongie S (2017) Feature Pyramid Networks for Object Detection. In ‘2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR)’, Honolulu, HI, USA, 21–26 July 2017. pp. 936–944. 10.1109/CVPR.2017.106
Liu W, Anguelov D, Erhan D, Szegedy C, Reed S, Fu CY, Berg AC (2016) SSD: Single shot multibox detector. In ‘Computer Vision - ECCV 2016. ECCV Lecture Notes in Computer Science ’. (Eds B Leibe, J Matas, N Sebe, M Welling) pp. 21–37. (Springer: Cham) 10.1007/978‐3‐319‐46448‐0_2
Liu NA, Lei J, Gao W, Chen HX, Xie XD (2021) Combustion dynamics of large-scale wildfires. Proceedings of the Combustion Institute 38(1), 157-198.
| Crossref | Google Scholar |
Ma N, Zhang X, Zheng HT, Sun J (2018) Shufflenet v2: Practical guidelines for efficient CNN architecture design. In ‘Computer Vision - ECCV 2018. ECCV 2018 Lecture Notes in Computer Science’. pp. 116–131. (Springer: Cham) 10.1007/978‐3‐030‐01264‐9_8
Muhammad K, Ahmad J, Lv Z, Bellavista P, Yang P, Baik SW (2019) Efficient Deep CNN-Based Fire Detection and Localization in Video Surveillance Applications. IEEE Transactions on Systems, Man, and Cybernetics: Systems 49(7), 1419-1434.
| Crossref | Google Scholar |
National Wildfire Coordinating Group (2020) Extreme Fire Behavior. Available at https://www.nwcg.gov/term/glossary/extreme-fire-behavior [verified 2020]
Nguyen DT, Nguyen TN, Kim H, Lee HJ (2019) A High-Throughput and Power-Efficient FPGA Implementation of YOLO CNN for Object Detection. IEEE Transactions on Very Large Scale Integration (VLSI) Systems 27(8), 1861-1873.
| Crossref | Google Scholar |
Pritam D, Dewan JH (2017) Detection of fire using image processing techniques with LUV color space. In ‘2017 2nd International Conference for Convergence in Technology (I2CT)’, Mumbai, India, 7–9 April 2017. pp. 1158–1162. 10.1109/I2CT.2017.8226309
Qiang XH, Zhou GX, Chen AB, Zhang X, Zhang WZ (2021) Forest fire smoke detection under complex backgrounds using TRPCA and TSVB. International Journal of Wildland Fire 30(5), 329-350.
| Crossref | Google Scholar |
Qin L, Shi Y, He YH, Zhang JR, Zhang XS, Li YJ, Deng T, Yan HM (2022) ID-YOLO: Real-Time Salient Object Detection Based on the Driver’s Fixation Region. IEEE Transactions on Intelligent Transportation Systems 23(9), 15898-15908.
| Crossref | Google Scholar |
Redmon J, Divvala S, Girshick R, Farhadi A (2016) You only look once: unified, real-time object detection. In ‘2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR)’, Las Vegas, NV, USA, 27–30 June 2016. pp. 779–788. 10.1109/CVPR.2016.91
Sadykova D, Pernebayeva D, Bagheri M, James A (2020) IN-YOLO: Real-Time Detection of Outdoor High Voltage Insulators Using UAV Imaging. IEEE Transactions on Power Delivery 35(3), 1599-1601.
| Crossref | Google Scholar |
Selvaraju RR, Cogswell M, Das A, Vedantam R, Parikh D, Batra D (2017) Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localization. In ‘2017 IEEE International Conference on Computer Vision (ICCV)’, Venice, Italy, 22–29 October 2017. pp. 618–626. 10.1109/ICCV.2017.74
Sharma LK, Gupta R, Fatima N (2022) Assessing the predictive efficacy of six machine learning algorithms for the susceptibility of Indian forests to fire. International Journal of Wildland Fire 31(8), 735-758.
| Crossref | Google Scholar |
Shees A, Ansari MS, Varshney A, Asghar MN, Kanwal N (2023) FireNet-v2: Improved Lightweight Fire Detection Model for Real-Time IoT Applications. Procedia Computer Science 218, 2233-2242.
| Crossref | Google Scholar |
Sudhakar S, Vijayakumar V, Sathiya Kumar C, Priya V, Ravi L, Subramaniyaswamy V (2020) Unmanned Aerial Vehicle (UAV) based Forest Fire Detection and monitoring for reducing false alarms in forest-fires. Computer Communications 149, 1-16.
| Crossref | Google Scholar |
Tedim F, Leone V, Amraoui M, Bouillon C, Coughlan MR, Delogu GM, Fernandes PM, Ferreira C, McCaffrey S, McGee TK, Parente J, Paton D, Pereira MG, Ribeiro LM, Viegas DX, Xanthopoulos G (2018) Defining Extreme Wildfire Events: Difficulties, Challenges, and Impacts. Fire 1(1), 9.
| Crossref | Google Scholar |
The People’s Government of Sichuan Province (2020) “3.30” Forest Fire Event Investigation Result in Xichang City, Liangshan Prefecture. Available at https://www.sc.gov.cn/10462/10464/13722/2020/12/21/76441b52bf034463946b09f61876d3f9.shtml [verified 21 December 2020]
Tohidi A, Gollner MJ, Xiao HH (2018) Fire Whirls. Annual Review of Fluid Mechanics 50, 187-213.
| Crossref | Google Scholar |
Viegas DX, Simeoni A (2011) Eruptive Behaviour of Forest Fires. Fire Technology 47, 303-320.
| Crossref | Google Scholar |
Wang T, Shi L, Yuan P, Bu LP, Hou XG (2017) A New Fire Detection Method Based on Flame Color Dispersion and Similarity in Consecutive Frames. In ‘2017 Chinese Automation Congress (CAC)’, Jinan, China, 20–22 October 2017. pp. 151–156. 10.1109/CAC.2017.8242754
Wang C-Y, Liao H-YM, Wu Y-H, Chen P-Y, Hsieh J-W, Yeh I-H (2020) CSPNet: A new backbone that can enhance learning capability of CNN. In ‘2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW)’, Seattle, WA, USA, 14–19 June 2020. pp. 1571–1580. 10.1109/CVPRW50498.2020.00203
Werth PA, Potter BE, Clements CB, Finney MA, Goodrick SL, Alexander ME, Cruz MG, Forthofer JA, McAllister SS (2011) Synthesis of knowledge of extreme fire behavior: Volume I for fire managers. General Technical Report PNW-GTR-54. (USDA Forest Service, Pacific Northwest Research Station: Portland, OR)
Werth PA, Potter BE, Alexander ME, Clements CB, Cruz MG, Finney MA, Forthofer JM, Goodrick SL, Hoffman C, Jolly WM, Mcallister SS, Ottmar RD, Parsons RA (2016) Synthesis of Knowledge of Extreme Fire Behavior: Volume 2 for fire behavior specialists, researchers, and meteorologists. General Technical Report PNW-GTR-891. (USDA Forest Service, Pacific Northwest Research Station: Portland, OR)
Wu XQ, Tan GH, Zhu NB, Chen ZL, Yang Y, Wen HX, Li KL (2021) CacheTrack-YOLO: Real-Time Detection and Tracking for Thyroid Nodules and Surrounding Tissues in Ultrasound Videos. Ieee Journal of Biomedical and Health Informatics 25(10), 3812-3823.
| Crossref | Google Scholar | PubMed |
Wu YF, Li JW, Bi S, Zhu X, Wang QF (2023) Research on Improved Ant Colony Algorithm for Mountain Hiking Emergency Rescue Path Planning. Journal of Geo-Information Science 25(1), 90-101.
| Crossref | Google Scholar |
Zaidi SSA, Ansari MS, Aslam A, Kanwal N, Asghar M, Lee B (2022) A survey of modern deep learning based object detection models. Digital Signal Processing 126, 103514.
| Crossref | Google Scholar |