NJUVISION/EDformer

EDformer

This is the source code for the ECCV 2024 - EDformer: Transformer-Based Event Denoising Across Varied Noise Levels.

Environment

Python 3.10
Pytorch 1.11.0
CUDA 11.3
cudnn 8

Installation

pip install -r requirements.txt

Require additional install sparseconvnet and pytorch3d

Datasets preparation

All datasets mentioned in the paper can be downloaded from BaiduDisk or OneDrive.

Train

python train.py [ED24 datasets root path] 4096

MESR Test

python eval_mesr.py

AUC Test

python eval_auc.py

zebra_fish Test

python eval_zebrafish.py