Skip to content

Unsupervised domain adaptation is critical in various computer vision tasks, such as object detection, instance segmentation, which aims to alleviate performance degradation caused by domain-shift and promotes the rapid application of models such as detection model in practical tasks. Previous works use adversarial training to align global featu…

Notifications You must be signed in to change notification settings

hustgetlijun/SC_DA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

29 Commits
 
 
 
 
 
 

Repository files navigation

SC_DA

Abstract:

Unsupervised domain adaptation is crucial for mitigating the performance degradation caused by domain bias in object detection tasks. Previous studies focus on pix-level and instance-level shifts alignment in attempts to minimize domain discrepancy. However, this method may lead to aligning single-class features with mixed-class features during image-level domain adaptation, given that each image in object detection tasks can belong to more than one category. To achieve the same category feature alignment between single-class and mixed-class, our method considers features with different mixed categories as a new class and proposes a mixed-classes $H-divergence$ to reduce domain bias for object detection. To enhance both single-class and mixed-class semantic information and achieve semantic separation for mixed-classes $H-divergence$, we employ Semantic Prediction Models (SPM) and Semantic Bridging Components (SBC). Furthermore, we reweigh the loss of the pix domain discriminator based on the SPM results to reduce sample imbalance. Our extensive experiments on widely used datasets illustrate how our method can robustly improve object detection in domain bias settings.

The Datasets of ours are:

https://pan.baidu.com/s/1ZiEdHgRVhmBZvywUhvbBkQ access code:odq6

pre-trained models:

https://pan.baidu.com/s/1yuwSkMjKSFr0InZAsHTs7Q access code: qhr8

Related works:

Main requirements:

  • torch == 1.0.0
  • torchvision == 0.2.0
  • Python 3

To train SW-ours on cityscape:

sh train_cityscape_allover.sh

To validate SW-ours on cityscape:

python test_cityscape_change.py

Environmental settings:

This repository is developed using python 3.6.7 on Ubuntu 16.04.5 LTS. The CUDA nad CUDNN version is 10.0 and 7.4.1 respectively. We use one NVIDIA 2080ti GPU card for training and testing. Other platforms or GPU cards such as V100 are tested.

Citing this repository:

About

Unsupervised domain adaptation is critical in various computer vision tasks, such as object detection, instance segmentation, which aims to alleviate performance degradation caused by domain-shift and promotes the rapid application of models such as detection model in practical tasks. Previous works use adversarial training to align global featu…

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published