基于改进YOLOv5的无人船视角目标检测方法

    Unmanned vessel viewpoint target detection method based on improved YOLOv5

    • 摘要: 针对复杂海况下无人船小尺度水面目标检测效果不佳及无人船数据集出现的样本不均衡的情况,本文提出一种改进YOLOv5的无人船水面目标检测方法。首先,将YOLOv5中主干网络部分替换为GhostNet轻量化网络。其次,在特征提取网络中加入全局空间自适应模块,以限制检测图空间维度上的影响。最后,损失函数部分基于多分类交叉熵损失函数加入可训练的噪声参数,降低样本不平衡带来的影响。通过无人船视角数据集试验结果表明,改进的方法mAP达到86.1%,相比于原始YOLOv5提升6.9%,网络的参数值降低43.4%,检测速度达到69.44帧数/秒,符合无人船等嵌入式设备的需求。改进的YOLOv5方法在检测无人船视角水面目标等情况,具有更好的检测效果。

       

      Abstract: Aiming at the poor detection of small-scale surface targets on unmanned vessels under complex sea conditions and the sample imbalance in the unmanned vessel dataset, an improved surface target detection method for unmanned vessels with YOLOv5 is proposed. Firstly, the backbone network in YOLOv5 is replaced by the GhostNet. Secondly, the global spatial adaptive module is added to the feature extraction network to limit the effect on the spatial dimension of the detection map. Finally, in order to reduce the impact of sample imbalances, trainable noise parameters are added to the loss function based on the multi-classification cross-entropy loss function. The calculation shows that the mAP value has reached 86.1%, which is 6.9% higher than the original YOLOv5.The parameter values have been reduced by 43.4%, and the detection speed reaches 69.44 frames/s, meeting the needs of embedded devices such as unmanned vessels. The improved YOLOv5 method has better detection effect in detecting water targets in unmanned vessel viewpoint.

       

    /

    返回文章
    返回