Tensor HDF5

class olympus.datasets.tensorhdf5.HDF5Dataset(data_path, transform=None, target_transform=None)[source]

Bases: torch.utils.data.dataset.Dataset

Dataset wrapping HDF5 tensors.

Attributes:
data
file
labels

Methods

register_datapipe_as_function  
register_function  
data
file
labels
olympus.datasets.tensorhdf5.generate_hdf5_dataset(file_name, shape=(3, 224, 224), num_class=1000, samples=192)[source]

Generate a Fake HDF5 Dataset for testing and benchmarking purposes