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

class cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.OneHotTransform(num_classes)[source]#

Bases: object

Methods

previous

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

next

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