Awesome
Image Lens Flare Removal Using Adversarial Curve Learning
This repository provides the official implementation for the following two papers:
Improving Lens Flare Removal with General Purpose Pipeline and Multiple Light Sources Recovery (ICCV 2023)
Image Lens Flare Removal Using Adversarial Curve Learning arxiv 2024
<img src="result-1.png" width="800px"/>Dataset
Training data
Wu el al. training images are provided in How to Train Neural Networks for Flare Removal (Wu et al. ICCV 2021)
Flare7K training images are provided in Flare7K: A Phenomenological Nighttime Flare Removal Dataset (Dai et al. NeurIPS 2022).
Please follow their instructions to access the data.
Test data
The consumer electronics dataset can be downloaded in Google Cloud or Baidu Netdisk (password:ptb3).
Pre-trained Model
The inference code based on Uformer can be downloaded in Google Cloud or Baidu Netdisk (password: vmxl).
Code
The backbone of our code is from Wu et al. How to Train Neural Networks for Flare Removal. We imporve the synthesis pipeline and light source recovery method to obtain better performance on flare removal.
Train
python train.py --flarec_dir=path/to/captured/flare --flares_dir=path/to/simulated/flare --scene_dir=path/to/scene/image
Test
python remove_flare.py --input_dir=path/to/test/image/dir --out_dir=path/to/output/dir --model=Uformer --batch_size=2 --ckpt=path/to/pretrained/model