Sep, 2023

UnitModule:用于水下对象检测的轻量级联合图像增强模块

TL;DR我们提出了一个插拔式的水下联合图像增强模块 (UnitModule),可以为检测器提供所需的输入图像,并且通过无监督学习提高 UnitModule 与检测器之间的交互效果。在水下图像的不同色偏情况下,UnitModule 通过配合的色偏损失和一种数据增强方法 (UCRT) 来改善性能。通过在不同的目标检测模型上进行广泛实验,我们发现 UnitModule 在 YOLOv5-S 模型上能够实现 2.6 AP 的最高性能提升,并可以在全新的测试集 (URPCtest) 上获得 3.3 AP 的提高。UnitModule 显著提高了我们测试的所有目标检测模型的性能,特别是对于参数数量较少的模型而言,而且对原始目标检测模型的推理速度影响较小。我们的定量和视觉分析也证明了 UnitModule 在增强输入图像和提高检测器对目标特征的感知能力方面的有效性。