基于改进Yolov8s的红外舰船目标检测算法

Infrared ship target detection algorithm based on improved Yolov8s

  • 摘要: 针对在复杂多变的海面上快速准确识别红外舰船的难题,提出了一种改进Yolov8s的红外舰船目标检测算法。首先采用轻量化网络EfficientViT(efficient vision transformer)作为主干网络,在确保精度的同时降低参数量和计算量;其次引入空间和通道重建卷积模块SCConv(spatial and channel reconstruction convolution)重构颈部网络C2f模块,减少模型参数冗余的同时,提高模型的检测精度;然后加入EMA(efficient multi-scale attention)注意力机制,提高模型的特征提取能力;最后使用WIoU(weighted intersection over union)损失函数代替CIoU(complete intersection over union)损失函数,进一步提升模型的泛化能力。实验结果表明,改进后的网络模型对比原Yolov8s网络模型,检测精度提升了2.5%,模型尺寸缩小了28%,参数量减少了31%,计算量降低了33%,在提升检测精度的同时使模型变得更加轻量化。改进后的Yolov8s网络模型在红外舰船检测任务上具有良好的性能。

     

    Abstract: Aiming at the difficulty of quickly and accurately recognizing infrared ships on the complex and changing sea surface, we proposed an infrared ship target detection algorithm with improved Yolov8s. Firstly, the lightweight network EfficientViT(efficient vision transformer) was used as the backbone network to reduce the number of parameters and computation while ensuring the accuracy.Secondly, the spatial and channel reconstruction convolution module SCConv(spatial and channel reconstruction convolution) was introduced to reconstruct the neck network C2f module, which could reduce the redundancy of the model parameters and at the same time improve the detection accuracy of the model. Then the EMA (efficient multi-scale attention) attention mechanism was incorporated to enhance the model's feature extraction capability.Finally the WIoU loss function was utilized to replace the CIoU(complete intersection over union) loss function to further enhance the model's generalization capability.The experimental results show that the improved network model improves the detection accuracy by 2.5%, reduces the model size by 28%, reduces the amount of parameters by 31%, and reduces the amount of computation by 33% compared with the original Yolov8s network model. The model becomes more lightweight while improving the detection accuracy. The improved Yolov8s network model has good performance on infrared ship detection tasks.

     

/

返回文章
返回