Xingang Xie, Ke Chen, Yiran Guo, Botao Tan, Lumeng Chen, Min Huang
{"title":"基于改进YOLOv5的火焰检测算法","authors":"Xingang Xie, Ke Chen, Yiran Guo, Botao Tan, Lumeng Chen, Min Huang","doi":"10.3390/fire6080313","DOIUrl":null,"url":null,"abstract":"Flame recognition is an important technique in firefighting, but existing image flame-detection methods are slow, low in accuracy, and cannot accurately identify small flame areas. Current detection technology struggles to satisfy the real-time detection requirements of firefighting drones at fire scenes. To improve this situation, we developed a YOLOv5-based real-time flame-detection algorithm. This algorithm can detect flames quickly and accurately. The main improvements are: (1) The embedded coordinate attention mechanism helps the model more precisely find and detect the target of interest. (2) We advanced the detection layer for small targets to enhance the model’s associated identification ability. (3) We introduced a novel loss function, α-IoU, and improved the accuracy of the regression results. (4) We combined the model with transfer learning to improve its accuracy. The experimental results indicate that the enhanced YOLOv5′s mAP can reach 96.6%, 5.4% higher than the original. The model needed 0.0177 s to identify a single image, demonstrating its efficiency. In summary, the enhanced YOLOv5 network model’s overall efficiency is superior to that of the original algorithm and existing mainstream identification approaches.","PeriodicalId":36395,"journal":{"name":"Fire-Switzerland","volume":null,"pages":null},"PeriodicalIF":3.0000,"publicationDate":"2023-08-13","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"1","resultStr":"{\"title\":\"A Flame-Detection Algorithm Using the Improved YOLOv5\",\"authors\":\"Xingang Xie, Ke Chen, Yiran Guo, Botao Tan, Lumeng Chen, Min Huang\",\"doi\":\"10.3390/fire6080313\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"Flame recognition is an important technique in firefighting, but existing image flame-detection methods are slow, low in accuracy, and cannot accurately identify small flame areas. Current detection technology struggles to satisfy the real-time detection requirements of firefighting drones at fire scenes. To improve this situation, we developed a YOLOv5-based real-time flame-detection algorithm. This algorithm can detect flames quickly and accurately. The main improvements are: (1) The embedded coordinate attention mechanism helps the model more precisely find and detect the target of interest. (2) We advanced the detection layer for small targets to enhance the model’s associated identification ability. (3) We introduced a novel loss function, α-IoU, and improved the accuracy of the regression results. (4) We combined the model with transfer learning to improve its accuracy. The experimental results indicate that the enhanced YOLOv5′s mAP can reach 96.6%, 5.4% higher than the original. The model needed 0.0177 s to identify a single image, demonstrating its efficiency. In summary, the enhanced YOLOv5 network model’s overall efficiency is superior to that of the original algorithm and existing mainstream identification approaches.\",\"PeriodicalId\":36395,\"journal\":{\"name\":\"Fire-Switzerland\",\"volume\":null,\"pages\":null},\"PeriodicalIF\":3.0000,\"publicationDate\":\"2023-08-13\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"1\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"Fire-Switzerland\",\"FirstCategoryId\":\"97\",\"ListUrlMain\":\"https://doi.org/10.3390/fire6080313\",\"RegionNum\":3,\"RegionCategory\":\"农林科学\",\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"Q2\",\"JCRName\":\"ECOLOGY\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"Fire-Switzerland","FirstCategoryId":"97","ListUrlMain":"https://doi.org/10.3390/fire6080313","RegionNum":3,"RegionCategory":"农林科学","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q2","JCRName":"ECOLOGY","Score":null,"Total":0}
A Flame-Detection Algorithm Using the Improved YOLOv5
Flame recognition is an important technique in firefighting, but existing image flame-detection methods are slow, low in accuracy, and cannot accurately identify small flame areas. Current detection technology struggles to satisfy the real-time detection requirements of firefighting drones at fire scenes. To improve this situation, we developed a YOLOv5-based real-time flame-detection algorithm. This algorithm can detect flames quickly and accurately. The main improvements are: (1) The embedded coordinate attention mechanism helps the model more precisely find and detect the target of interest. (2) We advanced the detection layer for small targets to enhance the model’s associated identification ability. (3) We introduced a novel loss function, α-IoU, and improved the accuracy of the regression results. (4) We combined the model with transfer learning to improve its accuracy. The experimental results indicate that the enhanced YOLOv5′s mAP can reach 96.6%, 5.4% higher than the original. The model needed 0.0177 s to identify a single image, demonstrating its efficiency. In summary, the enhanced YOLOv5 network model’s overall efficiency is superior to that of the original algorithm and existing mainstream identification approaches.