当前状态:
, 最新更新时间: ,
doi: 10.11999/JEIT251129
摘要:
针对现有水下目标检测方法在成像退化类型多样与背景干扰等复杂场景中鲁棒性差以及在设备资源受限条件下难以兼顾检测精度与模型轻量化的问题,该文提出一种基于冗余信息优化的轻量化水下目标检测网络(UnderWater Faster YOLO network based on redundancy information optimization, UWF-YOLO),并进一步构建了复杂场景水下目标检测数据集(Underwater Object Detection dataset with Complex Scene, CSUOD)。UWF-YOLO采用FasterNet Block重构C2f模块优化主干和颈部网络,通过特征通道选择机制减少冗余特征,并引入Ghost卷积增强颈部网络的多尺度特征融合能力;同时,通过基于分组卷积的参数共享检测头降低计算开销;最后,应用结构化通道剪枝技术进一步压缩网络规模。CSUOD数据集通过收集真实水下图像标注并进行分辨率标准化处理,覆盖雾化、色偏、非均匀照明等各种退化类型,可用于复杂场景下水下目标检测模型的鲁棒性训练与性能评测。在DUO,RUOD和TrashCan数据集上进行实验表明,相较于YOLOv8s,所提方法在计算量、权重大小与参数量3个指标上分别降低了60.4%、77.3%和78.4%;与参数量相当的YOLOv9-tiny相比,mAP指标在3个数据集上分别提升了0.3%、2.3%和3.4%。同时,在自建CSUOD数据集上的主客观对比实验,进一步证实所提模型在实现显著轻量化的同时,能够有效避免背景干扰导致的误检、漏检等问题,特别在复杂水下环境中展现出优异的检测性能。此外,该文构建的复杂场景水下数据集将有助于推动水下目标检测方法的发展。