cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.RemoveLabelTransform#

class cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.RemoveLabelTransform(remove_label, replace_with=0, input_key='seg', output_key='seg')[source]#

Bases: object

Replaces all pixels in data_dict[input_key] that have value remove_label with replace_with and saves the result to data_dict[output_key]

Methods