get_image_size

torchvision.transforms.functional.get_image_size(img: Tensor) → list[int]

以 [宽度, 高度] 的形式返回图像的大小。

参数:

img (PIL Image 或 Tensor) – 要检查的图像。

返回:

图像大小。

返回类型:

List[int]