cerebras.modelzoo.data.vision.segmentation.Hdf5BaseIterDataProcessor.Hdf5BaseIterDataProcessor#

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

Bases: abc.ABC, torch.utils.data.IterableDataset

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.Hdf5BaseIterDataProcessor

next

cerebras.modelzoo.data.vision.segmentation.Hdf5BaseIterDataProcessor.Hdf5BaseIterDataProcessorConfig