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].