MITD-YOLO:改进YOLOv8n的海上红外目标检测方法

MITD-YOLO: Maritime Infrared Target Detection Method Based on YOLOv8n

  • 摘要: 【目的】海上红外图像背景复杂、目标尺寸变化大、海浪杂波干扰,容易造成的目标漏检和误检问题,为提高红外图像中的目标检测准确率,提出了一种基于YOLOv8n的海上红外目标检测方法——MITD-YOLO(Maritime Infrared Target Detection-YOLO)。【方法】该方法引入了多样化分支模块(DBB)和多尺度卷积(EMSConv),采用三重注意力机制实现了空间和通道维度的特征交互,强化了关键特征提取,并且使用Powerful-IoUv2(PIoUv2)对原模型的损失函数进行了改进,提高检测精度并增强了模型的鲁棒性。【结果】实验结果表明,MITD-YOLO在海上红外目标检测精确率达到91.7%,比YOLOv8n模型提升了2.3%;召回率达到82.4%,提升了1.7%;平均精确率达到88.9%,提升了个202%;FPS达到132.8。【结论】该方法可提高海上红外目标检测效率。

     

    Abstract: Objective The complex backgrounds, significant variations in target sizes, and interference from ocean wave clutter in maritime infrared images often lead to missed and false detections. To enhance the accuracy of target detection in infrared images, a novel maritime infrared target detection method based on YOLOv8n, named MITD-YOLO (Maritime Infrared Target Detection-YOLO) was proposed. Methods The method integrates a Diversified Branch Block (DBB) and Enhanced Multi-Scale Convolution (EMSConv) modules. It employs a triple attention mechanism to facilitate feature interaction in spatial and channel dimensions, thereby strengthening the extraction of critical features. Additionally, the loss function of the original model is improved by incorporating Powerful-IoUv2 (PIoUv2), which enhances detection accuracy and robustness. Results Experimental results demonstrated that MITD-YOLO achieved a precision of 91.7% in maritime infrared target detection, representing a 2.3% improvement over the YOLOv8n model. The recall rate reached 82.4%, with an increase of 1.7%, and the average precision reaches 88.9%, marking an improvement of 20.2%. Furthermore, the method achieved a frame rate of 132.8 FPS. Conclusion The proposed method effectively improves the efficiency of maritime infrared target detection.

     

/

返回文章
返回