Preprocessing

utils.data.augmentation.get_random_pos(img, window_shape)

Extract of 2D random patch of shape window_shape in the image

utils.data.augmentation.pad_tensor(image_tensor: torch.Tensor, pad_size: int = 32)

Pads input tensor to make it’s height and width dividable by @pad_size

Parameters

• image_tensor – Input tensor of shape NxCxHxW

• pad_size – Pad size

Returns

Tuple of output tensor and pad params. Second argument can be used to reverse pad operation of metrics output

utils.data.augmentation.rm_pad_tensor(image_tensor, pad)

Remove padding from a tensor

Parameters

• image_tensor –

• pad –

Returns