岳旭生, 李军, 王耀弘, 等. 基于改进YOLOv5s的水面漂浮小目标检测算法[J]. 中国舰船研究, 2024, 19(X): 1–9. doi: 10.19693/j.issn.1673-3185.03689
引用本文: 岳旭生, 李军, 王耀弘, 等. 基于改进YOLOv5s的水面漂浮小目标检测算法[J]. 中国舰船研究, 2024, 19(X): 1–9. doi: 10.19693/j.issn.1673-3185.03689
YUE X S, LI J, WANG Y H, et al. A small floating target detection algorithm on the water surface based on the improved YOLOv5s[J]. Chinese Journal of Ship Research, 2024, 19(X): 1–9 (in Chinese). doi: 10.19693/j.issn.1673-3185.03689
Citation: YUE X S, LI J, WANG Y H, et al. A small floating target detection algorithm on the water surface based on the improved YOLOv5s[J]. Chinese Journal of Ship Research, 2024, 19(X): 1–9 (in Chinese). doi: 10.19693/j.issn.1673-3185.03689

基于改进YOLOv5s的水面漂浮小目标检测算法

A small floating target detection algorithm on the water surface based on the improved YOLOv5s

  • 摘要:
    目的 针对无人船视角下的水面漂浮瓶识别易出现错检、漏检等问题,基于YOLOv5s算法,提出一种改进的YOLOv5s水面漂浮小目标检测算法。
    方法 针对原始数据集Flow-Img进行了数据增强,对原始数据集进行扩充,从而避免模型出现过拟合的现象;为了提高深度学习模型对极小目标的检测精度,在YOLOv5s的3个检测层的基础上,增加了1个极小目标检测层,同时去掉用于大目标的检测头,避免数据不均衡带来的先验框分配问题;接着,在骨干网络中增加CBAM注意力模块,以解决模型在水面漂浮瓶检测任务中目标特征信息捕捉能力不足的问题;最后引入NWD回归损失函数,将IoU损失函数和NWD损失函数进行加权组合,形成一个综合的回归损失函数,从而进一步提高对水面漂浮瓶识别的准确率和精度。
    结果 实验结果表明,该算法在水面漂浮瓶检测上mAP@0.5值达到95.7%,对比原始的YOLOv5s算法,改进YOLOv5s算法的mAP@0.5提升了2.6%,mAP@0.95提升了4.5%,同时,模型参数量下降了61.9%。
    结论 在实现轻量化的同时使得水面漂浮瓶检测结果更加准确,为水面小型漂浮物的检测提供了重要的技术参考。

     

    Abstract: In order to solve the problems of false detection and missed detection in the recognition of floating bottles on the water surface from the perspective of unmanned boats, an improved YOLOv5s floating small target detection algorithm was proposed based on the YOLOv5s algorithm. Firstly, the data augmentation was carried out for the original dataset Flow-Img, and the original dataset was expanded to avoid the phenomenon of overfitting of the model. Secondly, in order to improve the detection accuracy of the deep learning model for very small targets, a very small target detection layer is added on the basis of the three detection layers of YOLOv5s, and the detection head for large targets is removed to avoid the problem of prior frame allocation caused by data imbalance. Thirdly, the CBAM attention module was added to the backbone network to solve the problem of insufficient ability of the model to capture target feature information in the floating bottle detection task on the water surface. Finally, the Normalized Wasserstein Distance (NWD) regression loss function is introduced, and the IoU loss function and the NWD loss function are weighted and combined to form a comprehensive regression loss function, so as to further improve the accuracy and precision of the recognition of floating bottles on the water surface. Experimental results show that the mAP@0.5 value of the algorithm reaches 95.7% in the detection of floating bottles on the water surface, and compared with the original YOLOv5s algorithm, the mAP@0.5 of the improved YOLOv5s algorithm is increased by 2.6%, the mAP@0.95 is increased by 4.5%, and the number of model parameters is reduced by 61.9%. While achieving lightweight, the detection results of floating bottles on the water surface are more accurate, which provides an important technical reference for the detection of small floating objects on the water surface.

     

/

返回文章
返回