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

Classes

MaskTransform

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