sherrycattt / evlowlight Goto Github PK

Paper | Project Page

Jinxiu Liang¹, Yixin Yang¹, Boyu Li¹, Peiqi Duan¹, Yong Xu², Boxin Shi¹

¹Peking University
²South China University of Technology

⭐If EvLowLight is helpful for you, please help star this repo. Thanks!🤗

• TODO
• Installation
• Inference
• Data Preparation

• Release inference code and pretrained models.
• Update links to paper and project page.
• Provide a runtime environment Docker image.
• Release train code and training set.

1. Clone this repo using git:

   git clone https://github.com/sherrycattt/EvLowLight.git

2. Create environment:

   Option 1: Using pip

   cd EvLowLight
   conda create -n evlowlight python=3.8
   conda activate evlowlight
   pip install -r requirements.txt

   Option 2: Using docker

   docker run --runtime=nvidia --gpus all --ipc=host --network=host  --rm -it \
     --ulimit memlock=-1 --ulimit stack=67108864 \
     -v `pwd`/EvLowLight:/workspace \
     -v `pwd`/timelens:/datasets/timelens \
     sherrycat/evlowlight

   Note the installation is only compatible with Linux users.

We provide an example for inference, check options/**_option.yml for more arguments.

python inference.py -opt options/timelens_option.yml

We provide example test data converted from the TimeLens for demo, which can be downloaded from Link (extracted code: Y9CN). Please place the dataset in the ../datasets folder. The dataset structure should be organized as follows:

├── timelens
│   └── events
│       ├── paprika_1000_gain_control_02
│       │   ├── events.txt
│       │   └── timestamp.txt
│       ├── pen_03
│       │   ├── events.txt
│       │   └── timestamp.txt
│       ...
│   └── low
│       ├── paprika_1000_gain_control_02
│       │   ├── 000000.png
│       │   └── 000001.png
│       │   ...
│       ├── pen_03
│       │   ├── 000000.png
│       │   └── 000001.png
│       │   ...
│       ...
│  ...

Each subfolder in the low folder contains image files with template filename %06d.png, and the file in the events subfolder contains events corresponding to the image subfolder with template filename events.txt defined as ev_file_ext in the option configuration file. Moreover, events also contains timestamp.txt where image timestamps are stored. The image stamps in timestamp.txt should match with the image files .

Please cite us if our work is useful for your research.

@inproceedings{liang2023evlowlight,
  title = {Coherent Event Guided Low-Light Video Enhancement},
  booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
  author = {Liang, Jinxiu and Yang, Yixin and Li, Boyu and Duan, Peiqi and Xu, Yong and Shi, Boxin},
  year = {2023},
  pages = {10615--10625},
}

This project is released under the Apache 2.0 license.

This project is based on BasicSR. Thanks for their awesome work.

If you have any questions, please feel free to contact with me at [email protected].