An improved YOLOv3-tiny method for fire detection in the construction industry

Abstract

To prevent fire accidents on construction site and improve the accuracy of fire detection, an improved YOLOv3-tiny method (I-YOLOv3-tiny) is proposed in this paper. Although the YOLOv3-tiny has a fast detection speed and low equipment requirement, the accuracy is relatively low on fire detection. The improvement of the I-YOLOv3-tiny method is followed by three steps. Firstly, the feature extraction of fire images is enhanced by optimizing the network structure. Secondly, a multi-scale fusion is used to improve the detection effect of fire targets. Finally, the anchor boxes that are suitable for fire data sets are selected by k-means clustering. The results show that I-YOLOv3-tiny has an increased percentage of 4 on the mAP, the Recall rate has an increased percentage of 4, and AVG IOU has an increased percentage of 6. The proposed model meets the real-time performance of fire detection. This study is of theoretical and practical significance on fire safety management and accident prevention in the construction industry.