matx.vision.tv_transforms package

class matx.vision.tv_transforms.CenterCrop(size: List[int], device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(size: List[int], device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.ColorJitter(brightness: List[float] = [0], contrast: List[float] = [0], saturation: List[float] = [0], hue: List[float] = [0], device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(brightness: List[float] = [0], contrast: List[float] = [0], saturation: List[float] = [0], hue: List[float] = [0], device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.Compose(device_id: int, transforms: List[Any])

Bases: object

__call__(imgs: Any) → Any

Call self as a function.

__init__(device_id: int, transforms: List[Any]) → None

class matx.vision.tv_transforms.ConvertImageDtype(dtype: str, global_scale: float = 1.0, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(dtype: str, global_scale: float = 1.0, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.CvtColor(color_code: str, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(color_code: str, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.Decode(to_rgb: bool = False, pool_size: int = 8, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(to_rgb: bool = False, pool_size: int = 8, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.GaussianBlur(kernel_size: List[int], sigma: List[float] = [0.1, 2.0], device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(kernel_size: List[int], sigma: List[float] = [0.1, 2.0], device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.Grayscale(num_output_channels: int = 1, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(num_output_channels: int = 1, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.Normalize(mean: List[float], std: List[float], global_scale: float = 1.0, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(mean: List[float], std: List[float], global_scale: float = 1.0, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.Pad(padding: List[int], fill: List[int] = [0], padding_mode: str = 'constant', device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(padding: List[int], fill: List[int] = [0], padding_mode: str = 'constant', device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomAdjustSharpness(sharpness_factor: float, p: float = 0.5, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(sharpness_factor: float, p: float = 0.5, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomAffine(degrees: List[float], translate: List[float] = [], scale: List[float] = [], shear: List[float] = [], interpolation: str = 'nearest', fill: List[float] = [], device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(degrees: List[float], translate: List[float] = [], scale: List[float] = [], shear: List[float] = [], interpolation: str = 'nearest', fill: List[float] = [], device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomAutocontrast(p: float = 0.5, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(p: float = 0.5, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomCrop(size: List[int], padding: List[int], pad_if_needed: bool = False, fill: List[int] = [0], padding_mode: str = 'constant', device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(size: List[int], padding: List[int], pad_if_needed: bool = False, fill: List[int] = [0], padding_mode: str = 'constant', device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomEqualize(p: float = 0.5, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(p: float = 0.5, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomGrayscale(p: float = 0.1, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(p: float = 0.1, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomHorizontalFlip(p: float = 0.5, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(p: float = 0.5, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomInvert(p: float = 0.5, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(p: float = 0.5, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomPerspective(distortion_scale: float = 0.5, interpolation: str = 'bilinear', fill: List[float] = [], p: float = 0.5, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(distortion_scale: float = 0.5, interpolation: str = 'bilinear', fill: List[float] = [], p: float = 0.5, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomPosterize(bits: int, p: float = 0.5, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(bits: int, p: float = 0.5, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomResizedCrop(size: List[int], scale: Tuple[float, float] = (0.08, 1.0), ratio: Tuple[float, float] = (0.75, 1.3333333333333333), interpolation: str = 'bilinear', device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(size: List[int], scale: Tuple[float, float] = (0.08, 1.0), ratio: Tuple[float, float] = (0.75, 1.3333333333333333), interpolation: str = 'bilinear', device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomRotation(degrees: List[float], interpolation: str = 'nearest', expand: bool = False, center: List[int] = [], fill: List[float] = [], device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(degrees: List[float], interpolation: str = 'nearest', expand: bool = False, center: List[int] = [], fill: List[float] = [], device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomSolarize(threshold: float, p: float = 0.5, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(threshold: float, p: float = 0.5, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.RandomVerticalFlip(p: float = 0.5, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(p: float = 0.5, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.Resize(size: List[int], interpolation: str = 'bilinear', resize_mode: str = '', max_size: int = 0, device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(size: List[int], interpolation: str = 'bilinear', resize_mode: str = '', max_size: int = 0, device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.Stack(device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(device_id: int = -2, sync: int = 0)

class matx.vision.tv_transforms.ToTensor(device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.Transpose(src_fmt: str = 'NHWC', dst_fmt: str = 'NCHW', device_id: int = -2, sync: int = 0)

Bases: BaseInterfaceClass

__call__(device: Any, device_str: str, sync: int) → Any

Call self as a function.

__init__(src_fmt: str = 'NHWC', dst_fmt: str = 'NCHW', device_id: int = -2, sync: int = 0) → None

class matx.vision.tv_transforms.set_device(device_id)

Bases: object

__init__(device_id)

Submodules

• matx.vision.tv_transforms.utils module
  • ToGpu
    • ToGpu.__call__()
    • ToGpu.__init__()
  • ToTorch
    • ToTorch.__call__()