get_image_size

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

以 [寬度, 高度] 的形式返回影像的大小。

引數:

img (PIL Image 或 Tensor) – 要檢查的影像。

返回:

影像大小。

返回型別:

List[int]