Skip to content

iCVTEAM/E2NeRF

Repository files navigation

Code for E2NeRF: Event Enhanced Neural Radiance Fields from Blurry Images (ICCV 2023)

Click here to see the video and supplementary materials in our project website.

Installation

The code is based on nerf-pytorch. Please refer to nerf-pytorch for the environment installation.

Dataset

Download the dataset here. The dataset contains "data" for training and "original data"

Data

Synthetic Data:

For the file of each scene, there are training images in the "train" file and the corresponding event data "events.pt". The ground truth images are in the "test" file.

Like in original NeRF, the training and testing poses are in the "transform_train.json" file and "transform_test.json" file. Notice that at the test time, we use the first pose of each view in "transform_test.json" to render the test images and the Ground Truth images are also rendered at this pose.

Real-World Data:

The structure is like original NeRF's llff data and the event data is in "event.pt".

Event Data:

For easy reading, we transform the event stream in to event bins as event.pt file. You can use pytorch to load the file. The shape of the tensor is (view_number, bin_number, H, W) and each element means the number of the events (positive and negative indicate polarity).

Original Data & Preproccesing

Synthetic Data:

There are original images for synthesizing the blurry image and the code. Besides, we supply the original event data generated from v2e. We also provide the code to transform the ".txt" event to "events.pt" for E2NeRF training.

Real-World Data:

We supply the original ".aedat4" data captured by davis346 and the processing code in the file. We also convert the event data into events.pt for training.

EDI:

We update the EDI code in the repository. You can use this code to deblur the images in the "train" file with corresponding events.pt data. And the deblurred images are saved at "images_for_colmap" file. Then, you can use colmap to generate the poses as in NeRF.

Releases

No releases published

Packages

No packages published

Languages