avalanche.benchmarks.datasets.Stream51

class avalanche.benchmarks.datasets.Stream51(root: ~typing.Optional[~typing.Union[str, ~pathlib.Path]] = None, *, train=True, transform=None, target_transform=None, loader=<function default_loader>, download=True)[source]

Stream-51 Pytorch Dataset

__init__(root: ~typing.Optional[~typing.Union[str, ~pathlib.Path]] = None, *, train=True, transform=None, target_transform=None, loader=<function default_loader>, download=True)[source]

Creates an instance of the Stream-51 dataset.

Parameters
  • root – The directory where the dataset can be found or downloaded. Defaults to None, which means that the default location for ‘stream51’ will be used.

  • train – If True, the training set will be returned. If False, the test set will be returned.

  • transform – The transformations to apply to the X values.

  • target_transform – The transformations to apply to the Y values.

  • loader – The image loader to use.

  • download – If True, the dataset will be downloaded if needed.

Methods

__init__([root, train, transform, ...])

Creates an instance of the Stream-51 dataset.

make_dataset(data_list[, ordering, seed])

data_list for train: [class_id, clip_num, video_num, frame_num, bbox, file_loc] for test: [class_id, bbox, file_loc]

Attributes