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