U-DECN: End-to-End Underwater Object Detection ConvNet with Improved DeNoising Training

TL;DR

U-DECN is a novel end-to-end underwater object detector that effectively handles underwater color cast noise, achieves high accuracy and speed, and is optimized for deployment on embedded underwater devices.

Contribution

The paper introduces U-DECN, a new ConvNet-based query end-to-end underwater object detector with specialized de-noising training and optimization for underwater environments.

Findings

Achieves 64.0 AP on DUO dataset.

Runs at 21 FPS on NVIDIA AGX Orin, 5 times faster than similar models.

Outperforms state-of-the-art query-based detectors in underwater scenarios.

Abstract

Underwater object detection has higher requirements of running speed and deployment efficiency for the detector due to its specific environmental challenges. NMS of two- or one-stage object detectors and transformer architecture of query-based end-to-end object detectors are not conducive to deployment on underwater embedded devices with limited processing power. As for the detrimental effect of underwater color cast noise, recent underwater object detectors make network architecture or training complex, which also hinders their application and deployment on unmanned underwater vehicles. In this paper, we propose the Underwater DECO with improved deNoising training (U-DECN), the query-based end-to-end object detector (with ConvNet encoder-decoder architecture) for underwater color cast noise that addresses the above problems. We integrate advanced technologies from DETR variants into DECO and design optimization methods specifically for the ConvNet architecture, including Deformable Convolution in SIM and Separate Contrastive DeNoising Forward methods. To address the underwater color cast noise issue, we propose an Underwater Color DeNoising Query method to improve the generalization of the model for the biased object feature information by different color cast noise. Our U-DECN, with ResNet-50 backbone, achieves the best 64.0 AP on DUO and the best 58.1 AP on RUOD, and 21 FPS (5 times faster than Deformable DETR and DINO 4 FPS) on NVIDIA AGX Orin by TensorRT FP16, outperforming the other state-of-the-art query-based end-to-end object detectors. The code is available at https://github.com/LEFTeyex/U-DECN.