Adverse Weather Image Translation with Asymmetric and Uncertainty-aware GAN (AU-GAN)
Official Tensorflow implementation of Adverse Weather Image Translation with Asymmetric and Uncertainty-aware GAN (AU-GAN)
Jeong-gi Kwak, Youngsaeng Jin, Yuanming Li, Dongsik Yoon, Donghyeon Kim and Hanseok Ko
British Machine Vision Conference (BMVC), 2021
Intro
Night → Day (BDD100K)
Rainy night → Day (Alderdey)
Architecture
Envs
git clone https://github.com/jgkwak95/AU-GAN.git
cd AU-GAN
# Create virtual environment
conda create -y --name augan python=3.6.7
conda activate augan
conda install tensorflow-gpu==1.14.0 # Tensorflow 1.14
pip install --no-cache-dir -r requirements.txt
Preparing datasets
Night → Day
Berkeley DeepDrive dataset contains 100,000 high resolution images of the urban roads for autonomous driving.
Rainy night → Day
Alderley dataset consists of images of two domains, rainy night and daytime. It was collected while driving the same route in each weather environment.
Please download datasets and then construct them following ForkGAN
Training
# Alderley (256x256)
python main_uncer.py --dataset_dir alderley
--phase train
--experiment_name alderley_exp
--batch_size 8
--load_size 286
--fine_size 256
--use_uncertainty True
# BDD100k (512x512)
python main_uncer.py --dataset_dir bdd100k
--phase train
--experiment_name bdd_exp
--batch_size 4
--load_size 572
--fine_size 512
--use_uncertainty True
Test
# Alderley (256x256)
python main_uncer.py --dataset_dir alderley
--phase test
--experiment_name alderley_exp
--batch_size 1
--load_size 286
--fine_size 256
# BDD100k (512x512)
python main_uncer.py --dataset_dir bdd100k
--phase test
--experiment_name bdd_exp
--batch_size 1
--load_size 572
--fine_size 512
Additional results
More results in paper and supplementary
Uncertainty map
Citation
If our code is helpful your research, please cite our paper:
@InProceedings{kwak_adverse_2021},
author = {Kwak, Jeong-gi and Jin, Youngsaeng and Li, Yuanming and Yoon, Dongsik and Kim, Donghyeon and Ko, Hanseok},
title = {Adverse Weather Image Translation with Asymmetric and Uncertainty-aware GAN},
booktitle = {British Conference of Computer Vision (BMVC)},
month = {November},
year = {2021}
}
Acknowledgments
Our code is bulided upon the ForkGAN implementation.
375 Dec 31, 2022
18 Dec 23, 2022
5 Jan 4, 2023
70 Dec 22, 2022
119 Dec 4, 2022
1 Feb 9, 2022
69 Nov 28, 2022
214 Jan 3, 2023
36 Dec 2, 2022
110 Dec 30, 2022
191 Dec 31, 2022
53 Nov 17, 2022
19 Dec 1, 2022
86 Dec 28, 2022
89 Dec 18, 2022
97 Jan 3, 2023
308 Jan 4, 2023
85 Dec 14, 2022
124 Dec 30, 2022