pytorch如何将图片转换为指定尺寸的张量（torch.Size([3, 602, 964])我想变成[3, 224, 224]）

在PyTorch中，如果你想要将图像转换为特定尺寸的张量，通常会先读取图像文件并利用`PIL`库进行预处理，然后再转换成`torch.Tensor`。下面是一个示例：

1. 导入必要的库[^1]:

import torchvision.transforms as transforms
from PIL import Image
import torch

2. 定义一个转换器，用于调整图像大小到所需的尺寸，如`(224, 224)`:

def resize_image(img_path, target_size):
    transform = transforms.Compose([
        transforms.Resize(target_size),  # 调整大小
        transforms.ToTensor(),  # 转换成张量
        transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),  # 归一化（可选）
    ])
    img = Image.open(img_path)
    return transform(img)

# 使用方法
img_path = "your_image.jpg"  # 替换为你的图片路径
resized_img = resize_image(img_path, (224, 224))
print(resized_img.shape)  # 应该输出 [3, 224, 224]

这里我们使用`transforms.Resize`来调整图像大小，`ToTensor()`将结果转换为PyTorch张量。如果需要归一化，可以添加`Normalize`变换。