The fire smoke detection algorithm based on improved YOLOv5s research

CAI Jing,ZHANG Zan,RAN Guangjin,LI Zhen,LI Liangrong
DOI: https://doi.org/10.3969/j.issn.2095-2163.2023.05.013
2023-01-01
Abstract:In order to solve the problems of false detection, missed detection and poor real-time performance in fire smoke detection algorithm, a fire smoke detection model based on yolov5s is proposed. Firstly, ghost revolution module is used to replace the conventional convolution module in the original yolov5s network structure to reduce the calculation cost and model parameters of the detection model on the basis of maintaining the same performance; Secondly, the vision transformer structure is added to the backbone network of the original yolov5s model to reduce the dependence on convolutional neural network and improve the ability to obtain global and local features; Finally, the coordinated attention mechanism is introduced to effectively extract feature information and further improve the accuracy of detection. The experimental results show that the parameters of the proposed fire smoke detection model are reduced by 18%, the accuracy is improved by 3.12%, and the detection speed is improved by 20%, which can meet the fire smoke detection in the actual scene.
What problem does this paper attempt to address?