torchvision.transforms.functional.hflip 是 PyTorch 中的一个函数,用于将输入的图像水平翻转。
该函数接受一个张量作为输入,表示要进行翻转的图像。如果输入是一个 PIL 图像,则需要先将其转换为张量。函数返回一个新的张量,其中包含水平翻转后的图像。
以下是一个示例代码:
import torchvision.transforms as transforms
from PIL import Image
# 从文件加载图像
image = Image.open("example.jpg")
# 将图像转换为张量
image_tensor = transforms.ToTensor()(image)
# 使用hflip水平翻转图像
hflip_image = transforms.functional.hflip(image_tensor)
# 将张量转换回 PIL 图像并显示
hflip_image = transforms.ToPILImage()(hflip_image)
hflip_image.show()