z-logo
open-access-imgOpen Access
YOLO-SF: YOLO for Fire Segmentation Detection
Author(s) -
Xianghong Cao,
Yixuan Su,
Xin Geng,
Yongdong Wang
Publication year - 2023
Publication title -
ieee access
Language(s) - English
Resource type - Journals
ISSN - 2169-3536
DOI - 10.1109/access.2023.3322143
Subject(s) - aerospace , bioengineering , communication, networking and broadcast technologies , components, circuits, devices and systems , computing and processing , engineered materials, dielectrics and plasmas , engineering profession , fields, waves and electromagnetics , general topics for engineers , geoscience , nuclear engineering , photonics and electrooptics , power, energy and industry applications , robotics and control systems , signal processing and analysis , transportation
Owing to the problems of missed detection, false detection, and low accuracy of the current fire detection algorithm, a segmentation detection algorithm, YOLO-SF, is proposed. This algorithm combines instance segmentation technology with the YOLOv7-Tiny object detection algorithm to improve its accuracy. We gather images that include both fire and non-fire elements to create a fire segmentation dataset (FSD). The segmentation detection head of YOLOR is adopted to improve the accuracy of model segmentation and enhance its ability to express details. The MobileViTv2 module is introduced to build the backbone network, which effectively reduces parameters while ensuring the network’s ability to extract features. The Efficient Layer Aggregation Network (ELAN) of the neck network is augmented with Convolutional Block Attention Module (CBAM) to broaden the receptive field of the model and enhance its attention to both the fire image channel and spatial information. Additionally, Varifocal Loss is used to address the problem of inaccurate object positioning in the edge areas of fire images. Compared with the YOLOv7-Tiny segmentation algorithm, for Box and Mask, the precision increases by 5.9% and 6.2%, recall increases by 2.5% and 3.3%, and mAP increases by 4% and 6%. In addition, the FPS reaches 55.64, satisfying the requirements for real-time detection. The improved algorithm exhibits good generalization performance and robustness.

The content you want is available to Zendy users.

Already have an account? Click here to sign in.
Having issues? You can contact us here