cerebras.modelzoo.data.vision.segmentation.InriaAerialDataProcessor.InriaAerialDataProcessor#

class cerebras.modelzoo.data.vision.segmentation.InriaAerialDataProcessor.InriaAerialDataProcessor(config)[source]#

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

Methods

create_dataloader

create_dataset

get_augment_transforms

get_tile_transform

preprocess_image

preprocess_mask

transform_image_and_mask

Attributes

tiling_image_shape

previous

cerebras.modelzoo.data.vision.segmentation.InriaAerialDataProcessor

next

cerebras.modelzoo.data.vision.segmentation.InriaAerialDataProcessor.InriaAerialDataProcessorConfig