<p>Fire detection is crucial for safeguarding human life and property. To address the limitations of existing deep learning-based detectors—such as weak feature perception, information loss, high computational cost, and poor performance on small targets—this paper proposes an enhanced YOLOv7 model named CGDS-YOLO. The model introduces three key innovations: a CDP-ELAN module (fusing Coordinate Convolution, Diverse Branch Block, and Partial Convolution) for strengthened feature extraction, a Gathering-Distributing mechanism for improved multi-scale information fusion, and a SlimNeck structure to reduce parameters while retaining fine-grained details. Additionally, Normalized Wasserstein Distance is adopted to enhance small target detection. Experiments on a homemade smoke and flame dataset and the public Visdrone dataset show that CGDS-YOLO outperforms baseline models, improving mAP by 2.0% and 1.7%, respectively, while maintaining high computational efficiency.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Enhanced YOLOv7 with CDP-ELAN and gather-distribute mechanism for robust smoke and flame detection

  • Junjie Hu,
  • Siyu He,
  • Yingjing Qian,
  • Qingli Zeng,
  • Yibo Niu,
  • Renmin Zhang

摘要

Fire detection is crucial for safeguarding human life and property. To address the limitations of existing deep learning-based detectors—such as weak feature perception, information loss, high computational cost, and poor performance on small targets—this paper proposes an enhanced YOLOv7 model named CGDS-YOLO. The model introduces three key innovations: a CDP-ELAN module (fusing Coordinate Convolution, Diverse Branch Block, and Partial Convolution) for strengthened feature extraction, a Gathering-Distributing mechanism for improved multi-scale information fusion, and a SlimNeck structure to reduce parameters while retaining fine-grained details. Additionally, Normalized Wasserstein Distance is adopted to enhance small target detection. Experiments on a homemade smoke and flame dataset and the public Visdrone dataset show that CGDS-YOLO outperforms baseline models, improving mAP by 2.0% and 1.7%, respectively, while maintaining high computational efficiency.