matx.vision.tv_transforms.utils module

class matx.vision.tv_transforms.utils.ToGpu(device_id: int = -2)[源代码]

基类:BatchBaseClass

__call__(imgs)[源代码]

Call self as a function.

__init__(device_id: int = -2)[源代码]
class matx.vision.tv_transforms.utils.ToTorch[源代码]

基类:object

__call__(imgs)[源代码]

Call self as a function.