IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v10y2022i7p1190-d787535.html
   My bibliography  Save this article

A Smoke Detection Model Based on Improved YOLOv5

Author

Listed:
  • Zhong Wang

    (School of Computer Science and Technology, Hefei Normal University, Hefei 230601, China)

  • Lei Wu

    (School of Computer Science and Technology, Hefei Normal University, Hefei 230601, China)

  • Tong Li

    (School of Computer Science and Technology, Hefei Normal University, Hefei 230601, China)

  • Peibei Shi

    (School of Computer Science and Technology, Hefei Normal University, Hefei 230601, China)

Abstract

Fast and accurate smoke detection is very important for reducing fire damage. Due to the complexity and changeable nature of smoke scenes, existing smoke detection technology has the problems of a low detection rate and a high false negative rate, and the robustness and generalization ability of the algorithms are not high. Therefore, this paper proposes a smoke detection model based on the improved YOLOv5. First, a large number of real smoke and synthetic smoke images were collected to form a dataset. Different loss functions (GIoU, DIoU, CIoU) were used on three different models of YOLOv5 (YOLOv5s, YOLOv5m, YOLOv5l), and YOLOv5m was used as the baseline model. Then, because of the problem of small numbers of smoke training samples, the mosaic enhancement method was used to randomly crop, scale and arrange nine images to form new images. To solve the problem of inaccurate anchor box prior information in YOLOv5, a dynamic anchor box mechanism is proposed. An anchor box was generated for the training dataset through the k-means++ clustering algorithm. The dynamic anchor box module was added to the model, and the size and position of the anchor box were dynamically updated in the network training process. Aiming at the problem of unbalanced feature maps in different scales of YOLOv5, an attention mechanism is proposed to improve the network detection performance by adding channel attention and spatial attention to the original network structure. Compared with the traditional deep learning algorithm, the detection performance of the improved algorithm in this paper was is 4.4% higher than the mAP of the baseline model, and the detection speed reached 85 FPS, which is obviously better and can meet engineering application requirements.

Suggested Citation

  • Zhong Wang & Lei Wu & Tong Li & Peibei Shi, 2022. "A Smoke Detection Model Based on Improved YOLOv5," Mathematics, MDPI, vol. 10(7), pages 1-13, April.
  • Handle: RePEc:gam:jmathe:v:10:y:2022:i:7:p:1190-:d:787535
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/10/7/1190/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/10/7/1190/
    Download Restriction: no
    ---><---

    Citations

    Citations are extracted by the CitEc Project, subscribe to its RSS feed for this item.
    as


    Cited by:

    1. Ssu-Han Chen & Jer-Huan Jang & Meng-Jey Youh & Yen-Ting Chou & Chih-Hsiang Kang & Chang-Yen Wu & Chih-Ming Chen & Jiun-Shiung Lin & Jin-Kwan Lin & Kevin Fong-Rey Liu, 2023. "Real-Time Video Smoke Detection Based on Deep Domain Adaptation for Injection Molding Machines," Mathematics, MDPI, vol. 11(17), pages 1-18, August.
    2. Hai Li & Peng Sun, 2023. "Image-Based Fire Detection Using Dynamic Threshold Grayscale Segmentation and Residual Network Transfer Learning," Mathematics, MDPI, vol. 11(18), pages 1-21, September.

    Corrections

    All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:10:y:2022:i:7:p:1190-:d:787535. See general information about how to correct material in RePEc.

    If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

    We have no bibliographic references for this item. You can help adding them by using this form .

    If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

    For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

    Please note that corrections may take a couple of weeks to filter through the various RePEc services.

    IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.