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.

Domain Generalisation for Object Detection under Covariate and Concept Shift

Joint Feature-Level And Pixel-Level Domain Adaption For Object Detection In The Wild

Domain Adaptation for Infrared-Radar Cross-Scene Multimodal Detection

Towards Domain Generalization in Object Detection

Domain-Invariant Proposals Based on a Balanced Domain Classifier for Object Detection

Domain Adaptive Faster R-CNN for Object Detection in the Wild

Improving Single Domain-Generalized Object Detection: A Focus on Diversification and Alignment

Domain Adaptive YOLO for One-Stage Cross-Domain Detection

Multilevel Knowledge Transfer for Cross-Domain Object Detection

Towards Effective Semantic OOD Detection in Unseen Domains: A Domain Generalization Perspective

Object-Aware Domain Generalization for Object Detection

Domain penalisation for improved Out-of-Distribution Generalisation

Domain-Invariant Disentangled Network for Generalizable Object Detection

Semi-Supervised Domain Generalization for Object Detection via Language-Guided Feature Alignment

Multi-Granularity Alignment Domain Adaptation for Object Detection

Generalize or Detect? Towards Robust Semantic Segmentation Under Multiple Distribution Shifts

Few-Shot Object Detection in Unseen Domains

Towards Domain Generalization for Multi-view 3D Object Detection in Bird-Eye-View

Domain Attention Model for Domain Generalization in Object Detection.

Domain Contrast for Domain Adaptive Object Detection

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