matx.vision.tv_transforms.utils module¶

class matx.vision.tv_transforms.utils.ToGpu(device_id: int = -2)[source]¶

Bases: BatchBaseClass

__call__(imgs)[source]¶: Call self as a function.

__init__(device_id: int = -2)[source]¶

class matx.vision.tv_transforms.utils.ToTorch[source]¶

Bases: object

__call__(imgs)[source]¶: Call self as a function.