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

class cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.OneHotTransformKits[source]#

Bases: object

Methods

__init__(num_classes)[source]#
__call__(data_dict)[source]#

Call self as a function.

previous

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

next

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