4KDehazing

This is the PyTorch implementation for our CVPR'21 paper. The model can removal hazy, smoke or even water impurities.

The repository includes:

1. Source code of our model.
2. Training code for O-hazy dataset.
3. Testing code for O-hazy dataset
4. Pre-trained model for O-hazy dataset.

Setup: 依赖的库 torch, numpy, tqdm, torchvision, kornia, opencv-python

Training 将带雾训练数据集放在./hazy 文件夹下 对应的清晰数据集放在./gt文件夹下。 运行命令 python train.py。 训练过程可在./result文件夹下找到。 模型保存在./model文件夹下。

Test model 将需要测试的数据集放在./OHAZE_test文件下。 运行命令 python test_model.py。 测试结果可在./test_result文件夹下找到。

Requirements： Python 3.7 PyTorch 1.6.0 CUDA 10.0 Ubuntu 16.04

Dataset Link：https://pan.baidu.com/s/1UqssfYFWI87r8ahFaqsTnQ Password：q2dn