ABSTRACT
To address the challenges posed by parametric and computationally intensive deep learning-based fire detection methods, hindering real-time detection solutions, we propose a lightweight fire detection algorithm named LWSCA-YOLOv5. Our method achieves strong detection performance with a significant reduction in FLOPs and Params. Firstly, we introduce a novel lightweight convolution module called SCAConvolution. This operation enables the embedding of object position information into the channel dimension during the feature fusion process. In comparison to standard convolution, SCAConvolution substantially reduces computational consumption while extracting richer features. Subsequently, we leverage this convolution unit to redesign the C3 module in YOLOv5, resulting in the proposed LWC3. To fully harness the potential of our method, we introduce the SPPAM module. Finally, we assemble the lightweight fire detection network using the aforementioned components, naming it LWSCA-YOLOv5. Experimental results demonstrate a 71% reduction in Params and a 63% reduction in FLOPs compared to the baseline while maintaining the same level of accuracy, validating the effectiveness of our method.
- Ren, S., He, K., Girshick, R., & Sun, J. 2015. Faster r-cnn: Towards real-time object detection with region proposal networks. Advances in neural information processing systems, 28.Google Scholar
- Redmon, J., Divvala, S., Girshick, R., & Farhadi, A. 2016. You only look once: Unified, real-time object detection. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 779-788).Google ScholarCross Ref
- Redmon, J., & Farhadi, A. 2017. YOLO9000: better, faster, stronger. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 7263-7271).Google ScholarCross Ref
- Redmon, J., & Farhadi, A. 2018. Yolov3: An incremental improvement. arXiv preprint arXiv:1804.02767.Google Scholar
- Bochkovskiy, A., Wang, C. Y., & Liao, H. Y. M. 2020. Yolov4: Optimal speed and accuracy of object detection. arXiv preprint arXiv:2004.10934.Google Scholar
- Wang, C. Y., Bochkovskiy, A., & Liao, H. Y. M. 2023. YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (pp. 7464-7475).Google ScholarCross Ref
- Liu, W., Anguelov, D., Erhan, D., Szegedy, C., Reed, S., Fu, C. Y., & Berg, A. C. 2016. Ssd: Single shot multibox detector. In Computer Vision–ECCV 2016: 14th European Conference, Amsterdam, The Netherlands, October 11–14, 2016, Proceedings, Part I 14 (pp. 21-37). Springer International Publishing.Google Scholar
- Barmpoutis, P., Dimitropoulos, K., Kaza, K., & Grammalidis, N. 2019, May. 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) (pp. 8301-8305). IEEE.Google ScholarCross Ref
- Chaoxia, C., Shang, W., & Zhang, F. 2020. Information-guided flame detection based on faster R-CNN. IEEE Access, 8, 58923-58932.Google ScholarCross Ref
- Wu, H., Wu, D., & Zhao, J. 2019. An intelligent fire detection approach through cameras based on computer vision methods. Process Safety and Environmental Protection, 127, 245-256.Google ScholarCross Ref
- Huo, Y., Zhang, Q., Jia, Y., Liu, D., Guan, J., Lin, G., & Zhang, Y. 2022. A deep separable convolutional neural network for multiscale image-based smoke detection. Fire Technology, 1-24.Google Scholar
- Yang, C., Pan, Y., Cao, Y., & Lu, X. 2022, September. CNN-Transformer Hybrid Architecture for Early Fire Detection. In International Conference on Artificial Neural Networks (pp. 570-581). Cham: Springer Nature Switzerland.Google Scholar
- Yan, C., Wang, Q., Zhao, Y., & Zhang, X. 2023. YOLOv5-CSF: an improved deep convolutional neural network for flame detection. Soft Computing, 1-11.Google Scholar
- Hou, Q., Zhou, D., & Feng, J. 2021. Coordinate attention for efficient mobile network design. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition (pp. 13713-13722).Google ScholarCross Ref
- Liu, Z., Lin, Y., Cao, Y., Hu, H., Wei, Y., Zhang, Z., ... & Guo, B. 2021. Swin transformer: Hierarchical vision transformer using shifted windows. In Proceedings of the IEEE/CVF international conference on computer vision (pp. 10012-10022).Google ScholarCross Ref
- Li, Y., Zhang, W., Liu, Y., Jing, R., & Liu, C. 2022. An efficient fire and smoke detection algorithm based on an end-to-end structured network. Engineering Applications of Artificial Intelligence, 116, 105492.Google ScholarDigital Library
- Howard, A. G., Zhu, M., Chen, B., Kalenichenko, D., Wang, W., Weyand, T., ... & Adam, H. 2017. Mobilenets: Efficient convolutional neural networks for mobile vision applications. arXiv preprint arXiv:1704.04861.Google Scholar
- Zhang, X., Zhou, X., Lin, M., & Sun, J. 2018. Shufflenet: An extremely efficient convolutional neural network for mobile devices. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 6848-6856).Google ScholarCross Ref
- Hu, J., Shen, L., & Sun, G. 2018. Squeeze-and-excitation networks. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 7132-7141).Google ScholarCross Ref
- Woo, S., Park, J., Lee, J. Y., & Kweon, I. S. 2018. Cbam: Convolutional block attention module. In Proceedings of the European conference on computer vision (ECCV) (pp. 3-19).Google ScholarDigital Library
- He, K., Zhang, X., Ren, S., & Sun, J. 2016. Deep residual learning for image recognition. In Proceedings of the IEEE conference on computer vision and pattern recognition (pp. 770-778).Google ScholarCross Ref
- He, K., Zhang, X., Ren, S., & Sun, J. 2015. Spatial pyramid pooling in deep convolutional networks for visual recognition. IEEE transactions on pattern analysis and machine intelligence, 37(9), 1904-1916.Google Scholar
- Wu, S., Zhang, X., Liu, R., & Li, B. 2023. A dataset for fire and smoke object detection. Multimedia Tools and Applications, 82(5), 6707-6726.Google ScholarDigital Library
Index Terms
- LWSCA-YOLOV5: An Improved Lightweight Fire Detection Algorithm Based on YOLOV5
Recommendations
Forest Fire Detection Based on Improved YOLOv5
AIPR '21: Proceedings of the 2021 4th International Conference on Artificial Intelligence and Pattern RecognitionTraditional forest fire detection based on sensing devices has a small range, low sensitivity and high false alarm rate, while existing forest fire detection methods based on deep learning have poor identification effect and inaccurate identification of ...
Research on Target Detection Method Based on Improved YOLOv5
Web Information Systems and ApplicationsAbstractAt present, various target detection algorithms are used in detection and classification. It is a problem to improve the accuracy and speed of target detection by using deep learning and neural network model to train the parameters of the data ...
Small Aerial Target Detection Algorithm Based on Improved YOLOv5
Advances in Swarm IntelligenceAbstractSmall target detection is a difficult problem in the field of target detection. Small targets have many characteristics, such as less visual information, difficult to extract discriminative features, easy background confusion and so on, as a ...
Comments