A Two-branch Neural Network for Non-homogeneous Dehazing via Ensemble Learning (https://arxiv.org/pdf/2104.08902.pdf)
- python3.7
- PyTorch >= 1.0
- NVIDIA GPU+CUDA
- numpy
- matplotlib
- tensorboardX(optional)
- Download ImageNet pretrained weights and our model weights.
- Download our dataset
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
@article{
}