shlogg · Early preview
Super Kai (Kazuya Ito) @superkai_kazuya

Transforming Images With FiveCrop In PyTorch

FiveCrop() crops an image into 5 parts. Size can be int or tuple/list(int) with 1 <= x. Use OxfordIIITPet() for dataset and torchvision.transforms.v2.FiveCrop() for transformation.

Buy Me a Coffee☕
*Memos:

My post explains OxfordIIITPet().

FiveCrop() can crop an image into 5 parts(Top-left, Top-right, Bottom-left, Bottom-right and Center) as shown below:
*Memos:

The 1st argument for initialization is size(Required-Type:int or tuple/list(int) or size()):
*Memos:

It's [height, width].
It must be 1 <= x.
A tuple/list must be the 1D with 1 or 2 elements.
A single value(int or tuple/list(int)) means [size, size].


The 1st argument is img(Required-Type:PIL Image or tensor(int)):
*Memos:

A tensor must be 2D or 3D.
Don't use img=.



v2 is recommended to use according to...