基于SSF-YOLOv7的森林火灾小目标检测算法

FOREST FIRE SMALL TARGET DETECTION ALGORITHM BASED ON SSF-YOLOV7

  • 摘要: 针对航拍森林火灾影像中的烟火目标尺寸较小导致其难以被检测的问题,提出一种森林火灾小目标检测算法——SSF-YOLOv7。设计一种高阶空间交互模块(C3HB)用于增强模型捕获烟火目标的全局信息的能力;加入无参注意力机制来提高烟火目标的可识别性,并使用K-means++聚类算法重新生成合适的锚框;将SAHI (Slice Aided Hyper Inference)技术融入模型,解决烟火目标细节不足的问题。实验结果表明,SSF-YOLOv7与YOLOv7相比,mAP₅₀提升了3.7百分点,烟火小目标的平均精确率提升了5.8百分点,FPS提升了19.8帧/s。

     

    Abstract: Aimed at the problem that the small size of pyrotechnic targets in aerial forest fire images makes them difficult to be detected, an improved forest fire small target detection algorithm (SSF-YOLOv7) is proposed. A high-order spatial interaction module (C3HB) was designed to enhance the model’s ability to capture the global information of pyrotechnic targets. A non-parametric attention mechanism was added to improve the recognizability of pyrotechnic targets, and K-means++ clustering algorithm was used to regenerate suitable anchor boxes. SAHI (Slice Aided Hyper Inference) technology was integrated into the model to solve the problem of insufficient details of pyrotechnic targets. The experimental results show that compared with YOLOv7, SSF-YOLOv7 has an increase of 3.7 percentage points in mAP₅₀, an increase of 5.8 percentage points in the average accuracy of pyrotechnic small targets, and an increase of 19.8 FPS.

     

/

返回文章
返回