Shortcuts

获取图像大小

torchvision.transforms.functional.get_image_size(img: Tensor) List[int][source]

返回图像的尺寸为[宽度, 高度]。

Parameters:

img (PIL ImageTensor) – 要检查的图像。

Returns:

图像大小。

Return type:

列表[int]