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

class cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.RenameTransform[source]#

Bases: object

Saves the value of data_dict[in_key] to data_dict[out_key]. Optionally removes data_dict[in_key] from the dict.

Methods

__init__(in_key, out_key, delete_old=False)[source]#
__call__(**data_dict)[source]#

Call self as a function.

previous

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

next

cerebras.modelzoo.data.vision.segmentation.transforms.utils