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

class cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.RemoveLabelTransform[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

__init__(remove_label, replace_with=0, input_key='seg', output_key='seg')[source]#
__call__(**data_dict)[source]#

Call self as a function.

previous

cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.OneHotTransformKits

next

cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.RenameTransform