Improved Flame Detection Based on Global Attention Mechanism and Deformable Convolution

Read the full article See related articles

Listed in

This article is not in any list yet, why not save it to one of your lists.
Log in to save this article

Abstract

This paper conducts a more thorough research and discussion on the detection of flame and smoke objects. To achieve real-time and accurate monitoring of smoke and fire targets in fire scenarios, the Improve-YOLOv8 algorithm is proposed based on YOLOv8. Firstly, this model replaces the C2f module with the DCNv2 module at the Backbone end, which can adaptively adjust according to the proportion and morphology of the flame and smoke targets. This design effectively addresses the issue of insufficient sampling of the fixed rectangular structure, thereby enhancing the models accuracy. Secondly, a small object detection layer is added in the Head part to address the challenge of accurately recognizing small flame targets. Additionally, the GAM attention mechanism is integrated after the Backbone SPPF module to further improve the accuracy and focus of image processing, thereby enhancing the overall processing efficiency. Finally, migration learning is implemented using sample data to effectively differentiate flame and smoke information and enhance detection accuracy. The experimental results show that the accuracy of the improved algorithm model reaches 98.6%, and the accuracy is improved by 10.5% compared with the original YOLOv8n, and mAP @0.5 9.6%. The comprehensive performance is improved, and the model can realize the real-time monitoring and accurate identification of fire and smoke.

Article activity feed