RepVGG-YOLOv7: A Modified YOLOv7 for Fire Smoke Detection

Autor: Xin Chen, Yipeng Xue, Qingshan Hou, Yan Fu, Yaolin Zhu
Jazyk: angličtina
Rok vydání: 2023
Předmět:
Zdroj: Fire, Vol 6, Iss 10, p 383 (2023)
Druh dokumentu: article
ISSN: 2571-6255
DOI: 10.3390/fire6100383
Popis: To further improve the detection of smoke and small target smoke in complex backgrounds, a novel smoke detection model called RepVGG-YOLOv7 is proposed in this paper. Firstly, the ECA attention mechanism and SIoU loss function are applied to the YOLOv7 network. The network effectively extracts the feature information of small targets and targets in complex backgrounds. Also, it makes the convergence of the loss function more stable and improves the regression accuracy. Secondly, RepVGG is added to the YOLOv7 backbone network to enhance the ability of the model to extract features in the training phase, while achieving lossless compression of the model in the inference phase. Finally, an improved non-maximal suppression algorithm is used to improve the detection in the case of dense smoke. Numerical experiments show that the detection accuracy of the proposed algorithm can reach about 95.1%, which contributes to smoke detection in complex backgrounds and small target smoke.
Databáze: Directory of Open Access Journals
Nepřihlášeným uživatelům se plný text nezobrazuje