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

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

next

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