arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.01293cs.CV

通过通道稳定化增强水下目标检测

Underwater Object Detection Enhancement via Channel Stabilization

Mohamed bin Zayed University of AI(穆罕默德·本·扎耶德人工智能大学)

Muhammad Ali, Salman Khan

AI总结:

本文提出通道稳定化与图像增强技术改善水下图像质量,并在TrashCan数据集上验证Detectron2检测性能,小目标平均精度提升9.53%。

摘要:

复杂的海洋环境极大地加剧了目标检测的挑战。海洋垃圾危害水生生态系统,构成一个持续性的难题。准确检测海洋沉积物对于减轻这种危害至关重要。我们的工作通过提升图像质量和评估检测方法来处理水下目标检测。我们使用Detectron2的主干网络以及多种基础模型和配置来完成此任务。我们提出了一种新颖的通道稳定化技术,并配合一个简化的图像增强模型,以减少训练图像中的雾霾和色偏,从而改善多尺度目标检测。在图像处理之后,我们测试了不同的Detectron2主干网络,以寻求最优检测精度。此外,我们应用了锐化滤波器和数据增强技术来突出目标轮廓,使其更易识别。结果在TrashCan数据集的实例版本和材质版本上进行了展示。性能最佳的主干网络方法结合了我们的通道稳定化和数据增强技术。我们还将Detectron2的检测结果与Deformable Transformer进行了比较。在TrashCan 1.0的实例版本中,与基线相比,我们的方法在小目标的平均精度上实现了9.53%的绝对提升,在边界框检测上实现了7%的绝对增益。代码将发布在Code: https://github.com/aliman80/Underwater-Object-Detection-via-Channel-Stablization。

英文摘要:

The complex marine environment exacerbates the challenges of object detection manifold. Marine trash endangers the aquatic ecosystem, presenting a persistent challenge. Accurate detection of marine deposits is crucial for mitigating this harm. Our work addresses underwater object detection by enhancing image quality and evaluating detection methods. We use Detectron2's backbone with various base models and configurations for this task. We propose a novel channel stabilization technique alongside a simplified image enhancement model to reduce haze and color cast in training images, improving multi-scale object detection. Following image processing, we test different Detectron2 backbones for optimal detection accuracy. Additionally, we apply a sharpening filter with augmentation techniques to highlight object profiles for easier recognition. Results are demonstrated on the TrashCan Dataset, both instance and material versions. The best-performing backbone method incorporates our channel stabilization and augmentation techniques. We also compare our Detectron2 detection results with the Deformable Transformer. In the instance version of TrashCan 1.0, our method achieves a 9.53% absolute increase in average precision for small objects and a 7% absolute gain in bounding box detection compared to the baseline. The code will be available on Code: https://github.com/aliman80/Underwater- Object-Detection-via-Channel-Stablization

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑