utils.datasets.PropheseeDatasetBase

class PropheseeDatasetBase(gt_files: List[str], data_files: List[str], time_step: int, num_load_file: int, name: str)

Bases: IterableDataset

Base class for Prophesee dataset iterators

Warning

This class can only be used as a base class for inheritance.

The samples_generator and parse_data method must be overridden in the child class.

Parameters:

• gt_files (List[str]) – List of ground truth file paths.

• data_files (List[str]) – List of data file paths.

• time_step (int) – Time between frames.

• num_load_file (int) – Number of examples loaded at a time.

• name (str) – The name of the dataset to download. Supported gen1 and 1mpx.

Raises:

ValueError – Invalid dataset name.

Methods

parse_data	Transforms events into a video stream
samples_generator	Creates a new sample generator

__iter__() → Iterator[Tuple[Tensor, Tensor]]

Returns an iterator for the dataset

parse_data(gt_boxes: Tensor, events_loader: PSEELoader) → Tuple[Tensor, Tensor]

Transforms events into a video stream

samples_generator() → Generator[Tuple[Tensor, Tensor], None, None]

Creates a new sample generator