sunxicauch / ntire-2021-dehazing-two-branch Goto Github PK

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.

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

• Download ImageNet pretrained weights and our model weights.
• Download our 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.

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

python test.py --model_save_dir results

Results on NTIRE 2021 NonHomogeneous Dehazing Challenge testing images:

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}
}