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

class cerebras.modelzoo.data.vision.segmentation.transforms.custom_transforms.MaskTransform(dct_for_where_it_was_used, mask_idx_in_seg=1, set_outside_to=0, data_key='data', seg_key='seg')[source]#

Bases: object

data[mask < 0] = 0 Sets everything outside the mask to 0. CAREFUL! outside is defined as < 0, not =0 (in the Mask)!!!

Parameters

  • dct_for_where_it_was_used –

  • mask_idx_in_seg –

  • set_outside_to –

  • data_key –

  • seg_key –

Methods

previous

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

next

cerebras.modelzoo.data.vision.segmentation.transforms.noise_augmentations