utils.datasets.MTPropheseeDataset

utils.datasets.MTPropheseeDataset#

class MTPropheseeDataset(gt_files: List[str], data_files: List[str], time_step: int, num_steps: int, num_load_file: int, name: str)[source]#

Bases: PropheseeDatasetBase

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
parse_data(gt_boxes: Tensor, events_loader: PSEELoader) → Tuple[Tensor, Tensor][source]#

Transforms events into a video stream

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

Creates a new sample generator