摘要:针对水下图像模糊、颜色失真, 水下场景环境复杂、目标特征提取能力有限等导致的水下鱼类目标检测精确度低的问题, 提出一种基于YOLOv5的改进水下鱼类目标检测算法. 首先, 针对水下图像模糊、颜色失真的问题, 引入水下暗通道优先 (underwater dark channel prior, UDCP)算法对图像进行预处理, 有助于在不同环境下正确识别目标; 然后, 针对水下场景复杂、目标特征提取能力有限问题, 在YOLOv5网络中引入高效的相关性通道 (efficient channel attention, ECA), 增强对目标的特征提取能力; 最后, 对损失函数进行改进, 提高目标检测框的准确度. 通过实验证明改进后的YOLOv5在水下鱼类目标检测中精确度比原始的YOLOv5提高了2.95%, 平均检测精度(mAP@0.5:0.95)提高了5.52%.