Abstract:The performance of existing underwater object detection methods severely degrades when they face the domain shift caused by complicated underwater environments. Due to the limited domain diversity in collected data, deep detectors easily memorize a few seen domains, which leads to low generalization ability. There are two common ideas to improve the domain generalization performance. First, it can be inferred that the detector trained on as many domains as possible is domain-invariant. Second, their hidden features should be equivalent because the images with the same semantic content are in different domains. This paper further excavates these two ideas and proposes a domain generalization framework that learns how to generalize across domains from Domain Mixup and Contrastive Learning (DMCL). First, based on the formation of underwater images, an image in one kind of underwater environment is the linear transformation of another underwater environment. Therefore, a style transfer model, which outputs a linear transformation matrix instead of the whole image, is proposed to transform images from one source domain to another, enriching the domain diversity of the training data. Second, the Mixup operation interpolates different domains on the feature level, sampling new domains on the domain manifold. Third, a contrastive loss is selectively applied to features from different domains to force the model to learn domain-invariant features but retain the discriminative capacity. With our method, detectors will be robust to domain shift. Also, a domain generalization benchmark S-UODAC2020 for detection is set up to measure the performance of our method. Comprehensive experiments on S-UODAC2020 and two object recognition benchmarks (PACS and VLCS) demonstrate that the proposed method is able to learn domain-invariant representations and outperforms other domain generalization methods. The code is available inhttps://github.com/mousecpn/DMC-Domain-Generalization-for-Underwater-Object-Detection.git.

WQT and DG-YOLO: towards domain generalization in underwater object detection

YoloXT: A Object Detection Algorithm for Marine Benthos

MDM-YOLO: Research on Object Detection Algorithm Based on Improved YOLOv4 for Marine Organisms.

Underwater Object Detection Based on Enhanced YOLO

Achieving domain generalization for underwater object detection by domain mixup and contrastive learning

Achieving Domain Generalization in Underwater Object Detection by Image Stylization and Domain Mixup.

An Attention Fusion Enhanced Underwater Object Detection Algorithm

UGC-YOLO: Underwater Environment Object Detection Based on YOLO with a Global Context Block

YOLO‐UOD: An underwater small object detector via improved efficient layer aggregation network

Underwater Object Detection Algorithm Based on an Improved YOLOv8

A Lightweight underwater detector enhanced by Attention mechanism, GSConv and WIoU on YOLOv8

Underwater Robot Target Detection Algorithm Based on YOLOv8

BG-YOLO: A Bidirectional-Guided Method for Underwater Object Detection

Investigating the Transferability of YOLOv5-Based Water Surface Object Detection Model in Maritime Applications

Image-Fused-Guided Underwater Object Detection Model Based on Improved YOLOv7

An Improved YOLOv5-Based Underwater Object-Detection Framework

ADOD: Adaptive Domain-Aware Object Detection with Residual Attention for Underwater Environments

Underwater Object Detection Using TC-YOLO with Attention Mechanisms

CEH-YOLO: A composite enhanced YOLO-based model for underwater object detection

YOLOv8-MU: An Improved YOLOv8 Underwater Detector Based on a Large Kernel Block and a Multi-Branch Reparameterization Module

Rethinking General Underwater Object Detection: Datasets, Challenges, and Solutions