get_image_size
-
torchvision.transforms.functional.get_image_size(img: torch.Tensor) → List[int][source] Returns the size of an image as [width, height].
- Parameters
img (PIL Image or Tensor) – The image to be checked.
- Returns
The image size.
- Return type
List[int]
