Skip to content

Official PyTorch implementation of Two-branch Dehazing, wining runner-up awards in NTIRE 2021.

License

Notifications You must be signed in to change notification settings

liuh127/NTIRE-2021-Dehazing-Two-branch

Repository files navigation

A Two-branch Neural Network for Non-homogeneous Dehazing via Ensemble Learning

This is the official PyTorch implementation of Two-branch Dehazing. See more details in [report] , [paper], [certificates]

Our method wins runner-up award in NTIRE 2021 Non-homogeneous Dehazing Challenge.

Dependencies and Installation

  • python3.7
  • PyTorch >= 1.0
  • NVIDIA GPU+CUDA
  • numpy
  • matplotlib
  • tensorboardX(optional)

Pretrained Weights & Dataset

Training set is the one that we used during the competition. Please be noted that we conduct gamma correction on NH-HAZE-2020. Test set is composed of random samples from the training set. The test set is used to provide effective training accuracy. If you want to obtain val and test accuracy on NH-HAZE-2021, please step towards the official competition server.

Train

python train.py --data_dir data -train_batch_size 8 --model_save_dir train_result

Test

python test.py --model_save_dir results

Qualitative Results

Results on NTIRE 2021 NonHomogeneous Dehazing Challenge testing images:

Citation

If you use any part of this code, please kindly cite

@InProceedings{Yu_2021_CVPR,
    author    = {Yu, Yankun and Liu, Huan and Fu, Minghan and Chen, Jun and Wang, Xiyao and Wang, Keyan},
    title     = {A Two-Branch Neural Network for Non-Homogeneous Dehazing via Ensemble Learning},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) Workshops},
    month     = {June},
    year      = {2021},
    pages     = {193-202}
}

About

Official PyTorch implementation of Two-branch Dehazing, wining runner-up awards in NTIRE 2021.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages