cerebras.modelzoo.data.vision.segmentation.CityscapesDataProcessor.CityscapesDataProcessor#
- class cerebras.modelzoo.data.vision.segmentation.CityscapesDataProcessor.CityscapesDataProcessor(params)[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