/NEST

Primary LanguagePython

NEST

NEST: Neural Event Stack for Event-based Image Enhancement, ECCV 2022

Minggui Teng, Chu Zhou, Hanyue Lou, and Boxin Shi.

Dependency

pip install -r requirements.txt

Dataset Folder

dataset
  ├─test
  │   ├─inputs  
  │   │       000_00000001.png
  │   │       ......   
  │   └─event
  │           000_00000001.npy
  │           ......
  └─train
      ├─inputs
      │       000_00000001.png
      │       ......     
      ├─event
      │       000_00000001.npy
      │       ......
      └─gt
              000_00000001.png
              ......

Train

Deblur

python main.py --train --mode deblur

SR

python main.py --train --mode sr

Test

Deblur

python main.py --mode deblur

SR

python main.py --mode sr

configurations can be changed in utils/options.py

Examples of HFR video generation application

examples1

More reuslts are in the subfolder results.

Contact

If you have any questions, please send an email to minggui_teng@pku.edu.cn