Study of Flame Detection based on Improved YOLOv4

Abstract

In some complex circumstances, the detection of conflagration mostly depends on smog detectors, which have lots of limitations in precision, efficiency and safety. If we make full use of object detection algorithms to detect the flame in industries, it will benefit people's safety obviously. Among all kinds of object detection algorithms, YOLO series play a very significant role. In this paper, we propose an improving strategy on YOLOv4 to enhance its precision based on multi-scale feature maps. Firstly, we create flame datasets including almost 4000 high-resolution flame pictures. Secondly, some improvements on feature extraction network are made to detect smaller objects. Finally, the total algorithm are trained and tested on our datasets for about 400 epochs. The result show that the method can generate high quality on flame detection in a great number of situations.