Home

Awesome

paper paper Weights PWC PWC PWC PWC

MixDehazeNet : Mix Structure Block For Image Dehazing Network

Abstract: Image dehazing is a typical task in the low-level vision field. Previous studies verified the effectiveness of the large convolutional kernel and attention mechanism in dehazing. However, there are two drawbacks: the multi-scale properties of an image are readily ignored when a large convolutional kernel is introduced, and the standard series connection of an attention module does not sufficiently consider an uneven hazy distribution. In this paper, we propose a novel framework named Mix Structure Image Dehazing Network (MixDehazeNet), which solves two issues mentioned above. Specifically, it mainly consists of two parts: the multi-scale parallel large convolution kernel module and the enhanced parallel attention module. Compared with a single large kernel, parallel large kernels with multi-scale are more capable of taking partial texture into account during the dehazing phase. In addition, an enhanced parallel attention module is developed, in which parallel connections of attention perform better at dehazing uneven hazy distribution. Extensive experiments on three benchmarks demonstrate the effectiveness of our proposed methods. For example, compared with the previous state-of-the-art methods, MixDehazeNet achieves a significant improvement (42.62dB PSNR) on the SOTS indoor dataset.

Framework:

image

News

March 15,2024: Paper accepted by IEEE IJCNN. The final version of this paper will be updated after the conference is completed on July 5.

Old Expriment Result:

image

Lastest Expriment Result (New Vision):

image image

About dataset:

Since my code refers to Dehazeformer, the dataset format is the same as that in Dehazeformer. In order to avoid errors when training the datasets, please download the datasets from Dehazeformer for training.

NH-Haze and Dense-Haze datasets (From Dehazeformer): https://drive.google.com/drive/folders/1NjLhzVZv2OZfs8DA6jMtGecLya_Gg9U_?usp=sharing

This is pretrain weights (google driver):

https://drive.google.com/drive/folders/1ep6W4H3vNxshYjq71Tb3MzxrXGgaiM6C?usp=drive_link

Help:

If you have any questions, you can send email to xiongqian2021@whut.edu.cn or xiongqian2019@outlook.com.

Thanks

Special Thanks to my supervisor and @IDKiro, they gave me selfless help in completing this work and answered my questions. Thank you very much.

Citation

If you find this work useful for your research, please cite our paper:

@inproceedings{lu2024mixdehazenet,
  title={Mixdehazenet: Mix structure block for image dehazing network},
  author={Lu, LiPing and Xiong, Qian and Xu, Bingrong and Chu, Duanfeng},
  booktitle={2024 International Joint Conference on Neural Networks (IJCNN)},
  pages={1--10},
  year={2024},
  organization={IEEE}
}