Object-Aware Domain Generalization for Object Detection

19 Dec 2023  ·  Wooju Lee, Dasol Hong, Hyungtae Lim, Hyun Myung ·

Single-domain generalization (S-DG) aims to generalize a model to unseen environments with a single-source domain. However, most S-DG approaches have been conducted in the field of classification. When these approaches are applied to object detection, the semantic features of some objects can be damaged, which can lead to imprecise object localization and misclassification. To address these problems, we propose an object-aware domain generalization (OA-DG) method for single-domain generalization in object detection. Our method consists of data augmentation and training strategy, which are called OA-Mix and OA-Loss, respectively. OA-Mix generates multi-domain data with multi-level transformation and object-aware mixing strategy. OA-Loss enables models to learn domain-invariant representations for objects and backgrounds from the original and OA-Mixed images. Our proposed method outperforms state-of-the-art works on standard benchmarks. Our code is available at https://github.com/WoojuLee24/OA-DG.

PDF Abstract

Results from the Paper

Task Dataset Model Metric Name Metric Value Global Rank Benchmark
Robust Object Detection Cityscapes OA-DG mPC [AP] 21.8 # 1
Robust Object Detection Cityscapes OA-Mix mPC [AP] 20.8 # 2
Robust Object Detection DWD OA-DG mPC [AP50] 31.8 # 1