cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms#
Classes
Utility function for pytorch. |
|
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] |
|
Saves the value of data_dict[in_key] to data_dict[out_key]. |