摘要:针对水下低质量成像、水下目标形态大小各异、以及水下目标重叠或遮挡导致水下目标检测精度低的问题,提出一种结 合数据增强和改进 YOLOv4(you look only once)的水下目标检测算法,在 YOLOv4 的主干特征提取网络 CSPDarknet53 中添加卷 积块注意力机制(convolutional block attention module,CBAM),以提高网络模型特征提取能力;在路径聚合网络( path aggregation network,PANet)中添加同层跳接和跨层跳接结构,以增强网络模型多尺度特征融合能力;通过数据增强方法 PredMix(predictionmix)模拟水下生物重叠、遮挡等显示不完全的情形,以增强网络模型鲁棒性。 实验结果表明,结合数据增强和改进 YOLOv4 的 水下目标检测算法在 URPC2018(underwater robot picking control 2018)数据集上的检测精度提升到了 78. 39%,比 YOLOv4 高出 7. 03%,充分证明所提算法的有效性。