search

jinyeying / FogRemoval

[ACCV22] Structure Representation Network and Uncertainty Feedback Learning for Dense Non-Uniform Fog Removal, https://arxiv.org/abs/2210.03061
149 stars 7 forks source link
deep-learning defog defogging dehaze dehazing dense smoke

readme

FogRemoval [ACCV'2022]

Introduction

Structure Representation Network and Uncertainty Feedback Learning for Dense Non-Uniform Fog Removal Asian Conference on Computer Vision (ACCV'2022)

[Paper] [Supp] arXiv [Poster] [Slides]

PWC PWC PWC

Datasets

1. SMOKE Dataset Dropbox | BaiduPan code:smok

${FogRemoval}
|-- Dataset_day
    |-- [Smoke](https://www.dropbox.com/home/badweather/ACCV2022_defog/Dataset_day/Smoke)
      |-- train (110 pairs)
         |-- hazy  
         |-- clean
      |-- [test] (12 pairs) 
         |-- hazy  
         |-- clean

SMOKE Dataset Dropbox BaiduPan code:smok
Fog Cityscapes Dropbox BaiduPan code:dfv2
Our SMOKE Results Dropbox BaiduPan code:goq0
Our Dense-HAZE Results Dropbox BaiduPan code:p1rn
Our NH-HAZE Results Dropbox BaiduPan code:i7fj
Our O-HAZE Results Dropbox BaiduPan code:7tzu

Pre-trained Model

Download the pre-trained NH-HAZE model Dropbox | BaiduPan code:hh81, put in results/NH-HAZE/model/NH-HAZE_params_0100000.pt

Test

python main_test.py --datasetpath [path_to_NH-HAZE dataset]

2. Fog Cityscapes Dropbox | BaiduPan code:dfv2

${FogRemoval}
|-- Dataset_day
    |-- Cityscapes
      |-- disparity 
      |-- leftImg8bit 
      |-- train (2,975 pairs)
         |-- hazy
         |-- clean 
      |-- test (1,525 pairs)
         |-- hazy  
         |-- clean 
      |-- generate_haze_cityscapes.m

Run the Matlab code to generate Synthetic Fog Cityscapes pairs:

Cityscapes/generate_haze_cityscapes.m

If smoke data is useful for your research, please cite our paper. 

@InProceedings{Jin_2022_ACCV,
    author    = {Jin, Yeying and Yan, Wending and Yang, Wenhan and Tan, Robby T.},
    title     = {Structure Representation Network and Uncertainty Feedback Learning for Dense Non-Uniform Fog Removal},
    booktitle = {Proceedings of the Asian Conference on Computer Vision (ACCV)},
    month     = {December},
    year      = {2022},
    pages     = {2041-2058}
}