cerebras.modelzoo.data.vision.segmentation.transforms.custom_transforms

Classes

MaskTransform

data[mask < 0] = 0 Sets everything outside the mask to 0.

previous

cerebras.modelzoo.data.vision.segmentation.transforms.crop_and_pad_augmentations.random_crop

next

cerebras.modelzoo.data.vision.segmentation.transforms.custom_transforms.MaskTransform