cerebras.modelzoo.data.vision.segmentation.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataProcessor#
- class cerebras.modelzoo.data.vision.segmentation.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataProcessor(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