Swin-YOLOv5: Research and Application of Fire and Smoke Detection Algorithm Based on YOLOv5

Abstract

Accurate monitoring of fire and smoke plays an irreplaceable role in preventing fires and safeguarding the safety of citizens' lives and property. The network structure of YOLOv5 is simple, but using convolution to extract features will lead to some problems such as limited receptive field, poor feature extraction ability, and insufficient feature integration. In view of the current defects of YOLOv5 target detection algorithm, a new algorithm model named Swin-YOLOv5 was proposed in this work. Swin transformation mechanism was introduced into YOLOv5 network, which enhanced the receptive field and feature extraction ability of the model without changing the depth of the model. In order to enrich the feature map splicing method of weighted Concat and enhance the feature fusion ability of model pairs, the feature splicing method of three output heads of feature fusion layer network was improved. The feature fusion module was further modified, and the weighted feature splicing method was introduced to improve the network feature fusion ability. Experimental results show that the map (average rage accuracy) of this method rises faster than the benchmark algorithm. Under the same experimental dataset, the map of this algorithm is improved by 0.7%, and the high-precision target detection speed is improved by 1.8 FPS (fast packet switch). Under the same experimental dataset, the improved algorithm could more accurately detect the targets that were not detected or detected inaccurately by the original algorithm, which embodied the adaptability of real scene detection and had practical significance. This work provided an opportunity for the application of fire-smoke detection in forest and indoor scenes and also developed a feasible idea for feature extraction and fusion of YOLOv5.