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