Lightweight Forest Flame Smoke Detection Algorithm Based on Yolov5
Forest smoke and flame detection is dominant in ensuring forest safety. To extinguish fire sources promptly and prevent the spread of wildfires, this paper proposes a lightweight improved YOLOv5 algorithm that is more accessible to embedded devices. The proposed algorithm is based on two pivotal ideas: (1) Introducing a normalization-based NAM attention mechanism into the neck network, which suppresses insignificant weights through weight sparsity penalties to enhance key feature extraction. (2) Incorporating a Slim-neck structure, which leverages GSConv and VoVGSCSP to construct a lightweight neck network. Research findings indicate that the optimized network outperforms the baseline YOLOv5s architecture, with a 3.34% gain in precision (P), while reducing FLOPs by 8.8%. This ensures a more lightweight model while maintaining detection accuracy.
References
- Pi, J., Liu, Y.H., Li, J.H. (2023) Research on lightweight forest fire detection algorithm based on YOLOv5s. Journal of Graphics, 44(01): 26-32.
- Yang, W., Yu, H.Y., Zhao, X.Y., et al. (2024) Forest fire detection algorithm based on reparameterized YOLOv5s. Radio Engineerinsg, 54(02): 284-293.
- Qian, C.S., Shen, Y.W., Sun, N., et al. (2023) Research on mountain fire detection method based on transformer improved YOLOv5. Electronic Measurement Technology, 46(16): 46-56.
- Li, J.W., Tang, H., Li, X.D., et al. (2024) LEF-YOLO: A lightweight method for intelligent detection of four extreme wildfires based on the YOLO framework. International Journal of Wildland Fire, 33(01).
- Li, H.L., Li, J., Wei, H.B., et al. (2022) Slim-neck by GSConv: A better design paradigm of detector architectures for autonomous vehicles. Journal of Real-Time Image Processing, 21.
- L, Y.C., Shao, Z.R., Teng, Y.Y., et al. (2021) NAM: Normalization-based Attention Module. arXiv preprint arXiv:2111.12419.
- Redmon, J., Farhadi, A. (2018) YOLOv3: An incremental improvement. arXiv preprint arXiv:1804.02767.