CWNet-Causal-Wavelet-Network - GitLite

KO | EN

gitlite — search

#javascript #python #hacktoberfest #go #rust #linux #java #awesome #cli #kubernetes #vue #pytorch #security #ios

CWNet-Causal-Wavelet-Network

(★ 98)

[ICCV 25] Official implementation for "CWNet: Causal Wavelet Network for Low-Light Image Enhancement"

파일 탐색기

최종 버전 다운로드 (.zip)

- __init__.py
- data_sampler.py
- LL_dataset.py
- LL_dataset_Invention.py
- util.py
- - __init__.py
  - arch_util.py
  - CWNet.py
  - ffc.py
  - layers.py
  - SS2D_arch.py
- __init__.py
- base_model.py
- enhancement_model.py
- loss.py
- lr_scheduler.py
- networks.py
- - LOL-v1.yml
  - LOL-v2-Real-Based_v1.yml
  - LOL-v2-Real.yml
  - LOL-v2-Syn.yml
  - LSRW-Huawei.yml
- - LOL-Syn.yml
  - LOL-v1.yml
  - LOL-v2-Real.yml
  - LSRW-Huawei.yml
- __init__.py
- options.py
- __init__.py
- util.py
environment.yml
LICENSE
README.md
test.py
train.py

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# [ICCV 25] CWNet: Causal Wavelet Network for Low-Light Image Enhancement [[Paper]](https://openaccess.thecvf.com/content/ICCV2025/html/Zhang_CWNet_Causal_Wavelet_Network_for_Low-Light_Image_Enhancement_ICCV_2025_paper.html)

- *Tongshun Zhang, Pingping Liu, Yubing Lu, Mengen Cai, Zijian Zhang, Zhe Zhang, Qiuzhan Zhou*
- *College of Computer Science and Technology, Jilin University*
- *Key Laboratory of Symbolic Computation and Knowledge Engineering of Ministry of Education*
- *College of Communication Engineering, Jilin University*

## 🔥 News

- [02/09/2025] We have updated our results based on the **LOL-v2-Real [[Google Drive]](https://drive.google.com/drive/folders/1Bcom7bANqh1_m2rNgEuG7C_JAAAF1bEh?usp=sharing)** training set and evaluated on the Real test set:

| Dataset | SSIM | PSNR | LPIPS |
|---------|------|------|-------|
| LOL-v2-Real (Updated) | 0.8604 | 21.65 | 0.0651 |
| Based on LOL-v1 (Previous/Paper) | 0.9005 | 27.39 | 0.0383 |

The previous results reported in our paper for LOL-v2-Real were based on the LOL-v1 dataset, which we refer to as "Based_v1".

## 1. Abstract
Traditional Low-Light Image Enhancement (LLIE) methods primarily focus on uniform brightness adjustment, often neglecting instance-level semantic information and the inherent characteristics of different features. To address these limitations, we propose CWNet (Causal Wavelet Network), a novel architecture that leverages wavelet transforms for causal reasoning. Specifically, our approach comprises two key components: 1) Inspired by the concept of intervention in causality, we adopt a causal reasoning perspective to reveal the underlying causal relationships in low-light enhancement. From a global perspective, we employ a metric learning strategy to ensure causal embeddings adhere to causal principles, separating them from non-causal confounding factors while focusing on the invariance of causal factors. At the local level, we introduce an instance-level CLIP semantic loss to precisely maintain causal factor consistency. 2) Based on our causal analysis, we present a wavelet transform-based backbone network that  effectively  optimizes the recovery of frequency information, ensuring precise enhancement tailored to the specific attributes of wavelet transforms. Extensive experiments demonstrate that CWNet significantly outperforms current state-of-the-art methods across multiple datasets, showcasing its robust performance across diverse scenes.

## CWNet Overall Architecture

![Over-all-Architecture](https://github.com/user-attachments/assets/2fea117c-a23e-44b5-b731-b19f931716a6)

## Performance

![Performance](https://github.com/user-attachments/assets/ef8b22c4-de69-4b70-a352-4f9951ac1798)

![Performance](https://github.com/user-attachments/assets/1d6a3968-60f4-441d-8e9a-bb80249f1b9a)

![Performance](https://github.com/user-attachments/assets/2d9b293c-3083-43fe-b90b-ea520da72fef)

## 2. Prepare Dataset

LOL-v1 [Baidu Disk](https://pan.baidu.com/share/init?surl=ZAC9TWR-YeuLIkWs3L7z4g&pwd=cyh2) (code: cyh2), [Google Drive](https://drive.google.com/file/d/1L-kqSQyrmMueBh_ziWoPFhfsAh50h20H/view?usp=sharing). From [Retinexformer](https://github.com/caiyuanhao1998/Retinexformer).

LOL-v2 [Baidu Disk](https://pan.baidu.com/share/init?surl=ZAC9TWR-YeuLIkWs3L7z4g&pwd=cyh2) (code: cyh2), [Google Drive](https://drive.google.com/file/d/1Ou9EljYZW8o5dbDCf9R34FS8Pd8kEp2U/view?usp=sharing). From [Retinexformer](https://github.com/caiyuanhao1998/Retinexformer).

LSRW-Huawei [Baidu Disk](https://pan.baidu.com/s/1XHWQAS0ZNrnCyZ-bq7MKvA) (code:wmrr). From [R2RNet](https://github.com/JianghaiSCU/R2RNet).

## 3. Create Environment

### Installation
```
conda env create -f environment.yml
conda activate cwnet
```
(1) First download the pre-trained model (DA-CLIP, Segmentation model) from [Google Drive](https://drive.google.com/drive/folders/1Bcom7bANqh1_m2rNgEuG7C_JAAAF1bEh?usp=sharing).

(2) Download the pre-trained model [DA-CLIP](https://github.com/Algolzw/daclip-uir).

(3) Download the pre-trained segmentation model hrnet_w48_pascal_context_cls59_480x480.pth, and configure line 21 of models/archs/segment/hrseg_model.py: pretrained_dict.

(4) Configure the path in /options/xx.yml:

lightmap_GT: Generate the light map path via dataroot_GT, generated by LIME ([LIME: Low-Light Image Enhancement via Illumination Map Estimation](https://ieeexplore.ieee.org/document/7782813).)
It can also be obtained from [Google Drive](https://drive.google.com/drive/folders/1Bcom7bANqh1_m2rNgEuG7C_JAAAF1bEh?usp=sharing).

dataroot_GT:  your image path

dataroot_LQ:  your image path

### 4. Train
```
python train.py -opt ./options/train/LSRW-Huawei.yml
```

### 5. Test
```
python test.py -opt ./options/test/LSRW-Huawei.yml
```

### 6. Pre-trained Models and Outputs
✨ [Google Drive](https://drive.google.com/drive/folders/1Bcom7bANqh1_m2rNgEuG7C_JAAAF1bEh?usp=sharing).

## Citation Information
If you find the project useful, please cite:

```bibtex  
@inproceedings{zhang2025cwnet,
  title={Cwnet: Causal wavelet network for low-light image enhancement},
  author={Zhang, Tongshun and Liu, Pingping and Lu, Yubing and Cai, Mengen and Zhang, Zijian and Zhang, Zhe and Zhou, Qiuzhan},
  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
  pages={8789--8799},
  year={2025}
}