YOLOV9-CBM: An Improved Fire Detection Algorithm Based on YOLOV9

Abstract

Regarding the current problems of false alarms and missed detections in fire detection, we propose a high-precision fire detection algorithm, YOLOV9-CBM (C3-SE, BiFPN, MPDIoU), by optimizing YOLOV9. Firstly, to tackle the shortage of both quality and quantity in the existing fire datasets, we collected 2,000 fire and smoke images to establish a dataset named CBM-Fire. Secondly, the RepNCSPELAN4 module of the YOLOv9 backbone was replaced with the C3 module containing SE Attention to improve detection efficiency while guaranteeing accuracy. Besides, we transformed the multi-scale fusion network PANet in the baseline algorithm into a bidirectional feature network pyramid BiFPN to facilitate the bidirectional flow of features, enabling the algorithm to fuse information at different scales more effectively. Finally, instead of CIoU losses, we adopted MPDIoU losses in bounding box regression, which improved the accuracy of model regression and classification. Experimental results indicate that compared with YOLOV9, the recall rate of YOLOV9-CBM has increased by 7.6% and the mAP has risen by 3.8%. The revised model demonstrates good generalization performance and robustness. Code and dataset are at https://github.com/GengHan-123/yolov9-cbm.git.