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

pip install -r requirements.txt

Require additional install sparseconvnet and pytorch3d

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

python train.py [ED24 datasets root path] 4096

python eval_mesr.py

python eval_auc.py

python eval_zebrafish.py