CUDA event warping and splatting

CUDA PyTorch extensions for iterative warping and splatting of events.

Ops

• iterative_3d_warp: iterative 3D warping of events following XY flow fields at regular Z intervals to these intervals
• trilinear_splat: trilinear splat to images at specified Z values

Install

pip install .
# or
pip install git+ssh://[email protected]/tudelft/cuda_event_ops

On Jetson, you can use jetson-containers make a container that can build the package:

jetson-containers build --name ceo cuda:12.2 pytorch:2.4