融合混合域注意力的YOLOv4在船舶检测中的应用

Application of YOLOv4 with Mixed-domain Attention in Ship Detection

海上船舶检测在海事监管领域发挥着重要的作用,然而由于海上的复杂环境以及船型的多样性,现有的基于卷积神经网络的方法在船舶检测领域难以同时满足高精度和实时的要求。针对复杂环境下海上船舶实时检测困难的问题,提出一种基于YOLOv4的YOLO-Marine模型,该模型将混合注意力机制引入检测网络的backbone部分,首先使用Mosaic方法对船舶数据进行预处理,然后通过K-Means++聚类得到初始anchor,并在Darknet上实现模型,用真实船舶数据集对模型进行训练和评估。实验结果表明YOLO-marine与YOLOv4相比,将船舶检测任务的mAP提升了2.1个百分点,在保证实时性的同时有效提高了船舶检测的精度,且在小目标和遮挡目标检测方面效果突出。

Marine ship detection plays an important role in the maritime field.Due to the complicated environment and the diversity of ships,existing methods based on convolutional neural network cannot achieve both high accuracy and real-time performance.To solve the problem of ship’s real-time detection in complicated environment,a YOLO-marine model based on YOLOv4 is proposed in which the domain attention mechanism is introduced into backbone.Firstly,the Mosaic method is used to preprocess the ship data.Then the K-Means++algorithm is used to get initial anchors.The model is implemented on Darknet for training and evaluation with the real ship dataset.The experimental result show that compared with YOLOv4,YOLO-marine improves the mAP of ship detection task by 2.1 percentage points.The model can effectively improve the accuracy of ship detection while ensuring real-time performance.It also gives outstanding results in small and occluded target.