draugr.torch_utilities.operations.torch_transforms.batch_transforms

Created on 17/07/2020

Classes

BatchToTensor()

Applies the ToTensor transform to a batch of images.

BatchNormalize(mean, std[, inplace, dtype, ...])

Applies the Normalize transform to a batch of images.

BatchRandomCrop(size[, padding, dtype, device])

Applies the RandomCrop transform to a batch of images. Args: size (int): Desired output size of the crop. padding (int, optional): Optional padding on each border of the image. Default is None, i.e no padding. dtype (torch.dtype,optional): The data type of tensors to which the transform will be applied. device (torch.device,optional): The device of tensors to which the transform will be applied.

BatchRandomHorizontalFlip([p, inplace])

Applies the RandomHorizontalFlip transform to a batch of images.