/Two-branch-dehazing

Primary LanguagePythonMIT LicenseMIT

A Two-branch Neural Network for Non-homogeneous Dehazing via Ensemble Learning (https://arxiv.org/pdf/2104.08902.pdf)

Dependencies and Installation

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

Pretrained Weights & Dataset

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

@article{
}