cerebras.modelzoo.data.vision.segmentation.Hdf5BaseDataProcessor.Hdf5BaseDataProcessor#

class cerebras.modelzoo.data.vision.segmentation.Hdf5BaseDataProcessor.Hdf5BaseDataProcessor(*args, **kwargs)[source]#

Bases: torch.utils.data.Dataset

A HDF5 dataset processor for UNet HDF dataset. Performs on-the-fly augmentation of image and labek.

Functionality includes:

Reading data from HDF5 documents Augmenting data

Methods

create_dataloader

Classmethod to create the dataloader object.
create_dataloader()[source]#

Classmethod to create the dataloader object.

previous

cerebras.modelzoo.data.vision.segmentation.Hdf5BaseDataProcessor

next

cerebras.modelzoo.data.vision.segmentation.Hdf5BaseDataProcessor.Hdf5BaseDataProcessorConfig