• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2022, Vol. 44 ›› Issue (12): 2196-2205.

• Graphics and Images • Previous Articles     Next Articles

Real-time flame detection with improved YOLO v4-tiny

WANG Guan-bo1,ZHAO Yi-fan2,LI Bo1,YANG Jun-dong1,DING Hong-wei1   

  1. (1.School of Information Science & Engineering,Yunnan University,Kunming 650504;
    2.School of Electrical and Information Engineering,Yunnan University for Nationalities,Kunming 650031,China)
  • Received:2021-08-23 Revised:2021-09-22 Accepted:2022-12-25 Online:2022-12-25 Published:2023-01-05

Abstract: In order to solve the problems of large number of parameters for real-time flame detection and high requirements for hardware computing power, a lightweight real-time flame detection model based on improved YOLO v4-tiny is proposed. Firstly, the parameters of the model are pruned. Secondly, by adding improved Receptive Field Blocks (CSP-RFBs) in the shallow layer of the model, the perceptual field of the model shallow layer is improved. Thirdly, the framework of CSP-ResNet is improved, and the “hourglass CSP-ResNet” with faster and higher accuracy is proposed. Finally, a modified Spatial Pyramid Pooling (SPP) is adopted at the deep level of the model to further fuse the multiple sensory fields. The experimental results show that the accuracy of the improved YOLO v4-tiny model can reach 48.5%, which is 15.5% better than the original model. The number of parameters of the model  and the weight size of the weight file are 2.45BFLOPs and 16.3Mb, which are 63.9% and 30.6% less than the original model, respectively. The FPS on the mobile development board NVIDIA Jeston Xavier can reach 49.6, which is 21.9% better than the original model.


Key words: lightweight object detection, YOLO v4-tiny, real-time flame detection, receptive field, hourglass CSP-ResNet