cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.NumpyToTensor#
- class cerebras.modelzoo.data.vision.segmentation.transforms.utility_transforms.NumpyToTensor(cast_to=None)[source]#
Bases:
object
Utility function for pytorch. Converts data (and seg) numpy ndarrays to pytorch tensors :param cast_to [list]: images will be cast to cast_to[0], targets will be cast to cast_to[1].
Methods