cerebras.modelzoo.data.vision.segmentation.Hdf5DataProcessor.Hdf5DataProcessor#

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

Bases: cerebras.modelzoo.data.vision.segmentation.Hdf5BaseDataProcessor.Hdf5BaseDataProcessor

Methods

create_dataloader

Classmethod to create the dataloader object.

get_augment_transforms

transform_image_and_mask

create_dataloader()#

Classmethod to create the dataloader object.

previous

cerebras.modelzoo.data.vision.segmentation.Hdf5DataProcessor

next

cerebras.modelzoo.data.vision.segmentation.Hdf5DataProcessor.Hdf5DataProcessorConfig