transforms.Compose 自定义输入

transforms.Compose 是 torchvision 库中的一个函数，用于将多个图像预处理操作组合在一起，以便对输入数据进行一系列的预处理。

你可以使用 transforms.Compose 自定义输入，按照自己的需求组合不同的图像预处理操作。以下是一个示例：

import torchvision.transforms as transforms

# 自定义预处理操作
custom_transforms = transforms.Compose([
    transforms.Resize((256, 256)),  # 调整图像大小为 256x256
    transforms.CenterCrop(224),  # 从中心裁剪出 224x224 的图像
    transforms.ToTensor(),  # 将图像转换为 Tensor 格式
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])  # 图像标准化
])

# 使用自定义预处理操作对输入数据进行处理
input_image = custom_transforms(input_image)

上述示例中，我们首先定义了一个自定义的预处理操作列表 `custom_transforms`，其中包括了对图像进行调整大小、中心裁剪、转换为 Tensor 格式和标准化的操作。然后，将输入图像 `input_image` 应用这些预处理操作。

通过使用 transforms.Compose，你可以根据自己的需求灵活地组合和定制图像预处理操作，以便符合你的任务需求。