Official PyTorch implementation of the paper "Exploring View Consistency for Scene-Adaptive Low-Light Light Field Image Enhancement", accepted by ICCV 2025 (High light).
Light field (LF) imaging in low-light environments suffers from severe noise and detail loss. While standard enhancement methods focus on individual views, they often break the View Consistency—the fundamental geometric relationship between sub-aperture images (SAIs).
Our method introduces a Scene-Adaptive framework that restores high-quality LF images while strictly preserving their 4D geometric structure, even in extreme darkness.
- Scene-Adaptive Luminance Mapping: Dynamically adjusts enhancement parameters based on global and local illumination statistics.
- Spatial-Angular Interaction (SAI) Module: Deeply integrates cross-view information to compensate for noise-corrupted pixels.
- EPI-based Geometric Constraint: A novel loss function targeting Epipolar Plane Images (EPI) to ensure smooth disparity and flicker-free view transitions.
The dataset used in this project is built upon the work from the following paper. If you use this dataset, please also cite their work:
Harnessing multi-view perspective of light fields for low-light imaging [l3fnet]
You can download our organized dataset from the following link:
- Baidu Netdisk (百度网盘): [L3F-data.zip]
- Extraction Code (提取码):
iaec
Please add the following citation when using the dataset
@ARTICLE{l3fnet,
author={M. {Lamba} and K. K. {Rachavarapu} and K. {Mitra}},
journal={IEEE Transactions on Image Processing},
title={Harnessing Multi-View Perspective of Light Fields for Low-Light Imaging},
year={2021},
volume={30},
pages={1501-1513},
doi={10.1109/TIP.2020.3045617}}This code was implemented and tested with Python and PyTorch. Based on the provided requirements, the main environment dependencies are:
- Python: >= 3.7 (Recommended 3.8 based on torch version)
- PyTorch: == 1.10.1
- Torchvision: == 0.11.2
To install all required dependencies, please run:
pip install -r requirements.txtTo train the model from scratch, you can use the following command:
python train.py --data_dir path/to/your/lowlightdata --output_dir path/to/save/outs --dataset all --epochs 10000 --learning_rate 1e-4 --gpu_no 0 --ckpt "" --patch 128 --n_view 5Training Arguments:
--data_dir: Directory path of the training dataset (default is).--output_dir: Output path for saving training results and model weights (default is).--dataset: Specifies the dataset used for training (default isall, meaning all data is used). It can be changed to1_20,1_50, or1_100.--epochs: Total number of training epochs (default is10000).--learning_rate: Initial learning rate of the model (default is1e-4).--gpu_no: Specifies the GPU ID to use (default is"0").--ckpt: Path to the checkpoint weight file for a pre-trained model or resuming training (default is empty"").--patch: Patch size for random image cropping during training (default is128).--n_view: Number of input views for the light field images (default is5).
To evaluate the trained model, run the following command:
python test.py --data_dir path/to/your/LowLightLF/ --output_dir path/to/save/outs --gpu_no 0 --patch 128 --n_view 5测试参数说明 (Testing Arguments):
Testing Arguments:
--data_dir: Directory path of the testing dataset (default is).--output_dir: Output path for saving testing results, such as the enhanced images (default is).--gpu_no: Specifies the GPU ID to use during testing (default is0).--patch: Patch size for image processing during testing (default is128).--n_view: Number of input views for the light field images during testing (default is5).
If you find our work useful, please cite:
@InProceedings{Zhang_2025_ICCV,
author = {Zhang, Shuo and Gao, Chen and Lin, youfang},
title = {Exploring View Consistency for Scene-Adaptive Low-Light Light Field Image Enhancement},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
year = {2025}
}For any questions, please contact Shuo Zhang at [zhangshuo@bjtu.edu.cn]. More research from our lab can be found at [INSIS-CV Page] .
