EDformer

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

Environment

Python 3.10
Pytorch 2.2.1
CUDA 11.8
cudnn 8
NVIDIA RTX 3090Ti

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