Skip to content

HazeSpace2M is a large scale paired hazy dataset for training and evaluating haze type classification models and single image dehazing models.

Notifications You must be signed in to change notification settings

tanvirnwu/HazeSpace2M

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

HazeSpace2M: A Dataset for Haze Aware Single Image Dehazing [Download]

Md Tanvir Islam 1, Nasir Rahim 1, Saeed Anwar 2, Muhammad Saqib 3, Sambit Bakshi 4, Khan Muhammad 1, *

| 1. Sungkyunkwan University, South Korea | 2. KFUPM, KSA | 3. UTS, Australia | 4. NIT Rourkela, India |

| *Corresponding Author |


---------- ## HazeSpace2M Dataset ![](./assets/HazeSpace2M.jpg) ## Haze Aware Dehazing ![](./assets/proposedFramework.jpg)

Dependencies

pip install -r requirements.txt

Dataset Download

We are preparing the complete dataset formatting with a structural naming convention. We will upload the full dataset as soon as we complete preparing the dataset with correct naming format as we displayed in the last images in this page.

Pretrained Weights

We will upload the pre-trained weights very soon.

Testing

python inference.py --gt_folder <path_to_gt> --hazy_folder <path_to_hazy> --output_dir <output_dir> --classifier <path_to_classifier> --cloudSD <path_to_cloudSD> --ehSD <path_to_ehSD> --fogSD <path_to_fogSD>

Note: Each variable is explained in the inference.py file.

Use Own Classifiers

To use your custom classifier, please follow the following steps:

  1. Write the code for your classifier architecture in the classifier.py file in the models folder.
  2. Now define the object of your classifier in the classification_inference method inside the conditionalDehazing.py file under the models folder.
  3. Finally, define the weights of your classifier inside the inference.py file

To use your custom specialized dehazers, please follow the following steps:

  1. Write the code for your classifier architecture in the dehazer.py file in the models folder.
  2. Now define the object of your dehazer in the load_model method inside the helper.py file under the utils folder.
  3. Finally, define the weights of your classifier inside the inference.py file

HazeSpace2M Folder Structure

Visitor Count

Cite this Paper

If you find our work useful in your research, please consider citing our paper:

@article{islam2024hazespace2m,
  title={HazeSpace2M: A Dataset for Haze Aware Single Image Dehazing},
  author={Islam, Md Tanvir and Rahim, Nasir and Anwar, Saeed and Saqib, Muhammad and Bakshi, Sambit and Muhammad, Khan},
  journal={arXiv preprint arXiv:2409.17432},
  year={2024}
}

@inproceedings{hazespace2m,
  title={HazeSpace2M: A Dataset for Haze Aware Single Image Dehazing},
  author={Islam, Md Tanvir and Rahim, Nasir and Anwar, Saeed and Saqib Muhammad and Bakshi, Sambit and Muhammad, Khan},
  booktitle={Proceedings of the 32nd ACM International Conference on Multimedia},
  year={2024},
  doi = {10.1145/3664647.3681382}
}