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

Classes

NumpyToTensor	Utility function for pytorch.
OneHotTransform
OneHotTransformKits
RemoveLabelTransform	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]
RenameTransform	Saves the value of data_dict[in_key] to data_dict[out_key].

previous

cerebras.modelzoo.data.vision.segmentation.transforms.resample_transforms.SimulateLowResolutionTransform

next

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