ES-YOLO: a new lightweight fire detection model

Shi Wang; XiaoHong Wang

doi:10.1117/12.2660040

3 February 2023 ES-YOLO: a new lightweight fire detection model

Shi Wang, XiaoHong Wang

Proceedings Volume 12511, Third International Conference on Computer Vision and Data Mining (ICCVDM 2022); 125111F (2023) https://doi.org/10.1117/12.2660040
Event: Third International Conference on Computer Vision and Data Mining (ICCVDM 2022), 2022, Hulun Buir, China

Abstract

For the problems of fire detection models based on computer vision, such as long inference and training time, too many model parameters and low detection accuracy. We propose ES-YOLO, which can quickly and accurately detect flames and smoke. Firstly, the original YOLOv5s backbone network is replaced with EfficientNetV2, which reduces the computational complexity of the network and improves the detection accuracy. Secondly, replaces the CIoU loss function with SIoU, which speeds up the convergence of the model. Finally, 9-Mosaic data augmentation is designed to enrich the dataset. The experimental results on the PASCAL VOC2007 dataset demonstrate that the mAP@0.5 and recall of ESYOLO are 20% and 15% higher than that of YOLOv5s, the size of the model are compressed to 1/2 of that of YOLOv5s. ES-YOLO meets the requirements of lightweight and real-time detection.

Citation Download Citation

Shi Wang and XiaoHong Wang "ES-YOLO: a new lightweight fire detection model", Proc. SPIE 12511, Third International Conference on Computer Vision and Data Mining (ICCVDM 2022), 125111F (3 February 2023); https://doi.org/10.1117/12.2660040

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available