https://github.com/cao-cong/ADS-SemiSeg

Adversarial Dual-Student with Differentiable Spatial Warping for Semi-Supervised Semantic Segmentation. TCSVT 2022
https://github.com/cao-cong/ADS-SemiSeg

Last synced: about 1 month ago
JSON representation

Adversarial Dual-Student with Differentiable Spatial Warping for Semi-Supervised Semantic Segmentation. TCSVT 2022

• Host: GitHub
• URL: https://github.com/cao-cong/ADS-SemiSeg
• Owner: cao-cong
• Created: 2022-09-11T11:54:00.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2023-11-19T17:51:28.000Z (about 1 year ago)
• Last Synced: 2024-08-03T01:11:52.086Z (5 months ago)
• Language: Python
• Homepage:
• Size: 357 KB
• Stars: 6
• Watchers: 2
• Forks: 1
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• Awesome-Semi-Supervised-Semantic-Segmentation - Code - cong/ADS-SemiSeg)|[Paper](https://arxiv.org/pdf/2203.02792.pdf)| (2022)

README

        
# ADS-SemiSeg

This repository contains official implementation of Adversarial Dual-Student with Differentiable Spatial Warping for Semi-Supervised Semantic Segmentation in TCSVT 2022, by Cong Cao, Tianwei Lin, Dongliang He, Fu Li, Huanjing Yue, Jingyu Yang, and Errui Ding. [[arxiv]](https://arxiv.org/abs/2203.02792) [[journal]](https://ieeexplore.ieee.org/abstract/document/9889741)



  



## Code

### Environment

- Python >= 3.5

- Pytorch >= 1.1

- NVIDIA Tesla V100

### Test

You can download pretrained weights from [here](https://drive.google.com/drive/folders/1Ch9bUbqToN2hisl3afnCW32qhP12p9SB?usp=sharing) (ADS-DGW_Dataset_SemiRatio_iterXXXXX.pth), then run:

```

bash run_scripts/test_VOC2012.sh

```

### Train

Train baseline:

```

bash run_scripts/train_baseline_VOC2012.sh

```

Train Mean-Teacher with DGW augmentation:

```

bash run_scripts/train_MT_DGW_VOC2012.sh

```

Train ADS with DGW augmentation:

```

bash run_scripts/train_ADS_DGW_VOC2012.sh

```

## Citation

If you find our paper or code helpful in your research or work, please cite our paper:

```

@article{cao2022adversarial,

  title={Adversarial dual-student with differentiable spatial warping for semi-supervised semantic segmentation},

  author={Cao, Cong and Lin, Tianwei and He, Dongliang and Li, Fu and Yue, Huanjing and Yang, Jingyu and Ding, Errui},

  journal={IEEE Transactions on Circuits and Systems for Video Technology},

  volume={33},

  number={2},

  pages={793--803},

  year={2022},

  publisher={IEEE}

}

```