Crop the given image into four corners and the central crop. This transform returns a tuple of images and there may be a mismatch in the number of inputs and targets your Dataset returns.

transform_five_crop(img, size)

Arguments

img

A magick-image, array or torch_tensor.

size

(integer vector or integer): Desired output size. If size is an integer vector of length 2 like c(h, w), output size will be matched to this. If size is a bare integer, smaller edge of the image will be matched to this number, i.e., if height > width, then image will be rescaled to (size * height / width, size).