avalanche.benchmarks.datasets.Stream51

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

Stream-51 Pytorch Dataset

__init__(root: ~pathlib.Path | str | None = None, *, train=True, transform=None, target_transform=None, loader=<function default_loader>, download=True)

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

root	The path to the dataset.
download	If True, the dataset will be downloaded (only if needed).
verbose	If True, some info about the download process will be printed.