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