基于改进YOLOv8的红外小目标检测算法研究

Research on Infrared Small Target Detection Algorithm Based on Improved YOLOv8

  • 摘要: 针对复杂背景下红外小目标识别错误率高,模型回归损失较大的问题,提出一种改进的算法YOLOv8_SG(Small goals)。通过构建小目标检测层、引入SA注意力机制与WIoU_v3损失函数,使算法能够融合更深层特征,具有更大的感受野,并且降低了训练样本标注质量不均衡的影响,提高了预测框的位置精度,增强了对小目标的检测能力。实验结果表明,改进后的算法mAP由0.8514提升到0.8997,Box_loss效果较改进前提升了34.9%,该算法在小目标检测上具有更高的特征提取能力和更高的检测精度。

     

    Abstract: Aiming at the problem of the high error rate of infrared small-target recognition and the large loss of model regression in complex backgrounds, an improved YOLOv8_SG (Small goals) algorithm was proposed by adding a small target detection layer and introducing the SA attention mechanism and WIoU_v3 loss function, which can fuse deeper features and have a larger receptive field. Moreover, the influence of the uneven labeling quality of the training samples was reduced, the position accuracy of the prediction box was improved, and the ability to detect small targets was enhanced. The experimental results show that the mAP of the improved algorithm increased from 0.8514 to 0.8997, and the overall loss effect of Box_loss increased by 34.9%. The proposed algorithm has a higher feature extraction ability and higher detection accuracy for small-target detection.

     

/

返回文章
返回