摘要
Due to the rapid growth and spread of fire,it poses a major threat to human life and property.Timely use of fire detection technology can reduce disaster losses.The traditional threshold segmentation method is unstable,and the flame recognition methods of deep learning require a large amount of labeled data for training.In order to solve these problems,this paper proposes a new method combining You Only Look Once version 5(YOLOv5)network model and improved flame segmentation algorithm.On the basis of the traditional color space threshold segmentation method,the original segmentation threshold is replaced by the proportion threshold,and the characteristic information of the flame is maximally retained.In the YOLOv5 network model,the training module is set by combining the ideas of Bootstrapping and cross validation,and the data distribution of YOLOv5 network training is adjusted.At the same time,the feature information after segmentation is added to the data set.Different from the training method that uses large-scale data sets for model training,the proposed method trains the model on the basis of a small data set,and achieves better model detection results,and the detection accuracy of the model in the validation set reaches 0.96.Experimental results show that the proposed method can detect flame features with faster speed and higher accuracy compared with the original method.
基金
supported by Hainan Natural Science Foundation of China(No.620RC602)
National Natural Science Foundation of China(No.61966013,12162012)
Hainan Provincial Key Laboratory of Ecological Civilization and Integrated Land-sea Development.
