利用pytorch的transforms对图片处理并显示的代码
时间: 2024-04-30 11:20:17 浏览: 201
pytorch中的transforms模块实例详解
你可以使用以下代码来利用PyTorch的transforms对图片进行处理并显示:
```python
import torch
import torchvision.transforms as transforms
from PIL import Image
# 定义transforms
transform = transforms.Compose([
transforms.Resize(256),
transforms.CenterCrop(224),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225])
])
# 读取图片
img = Image.open("example.jpg")
# 对图片进行处理
img = transform(img)
# 显示处理后的图片
import matplotlib.pyplot as plt
plt.imshow(img.permute(1, 2, 0))
plt.show()
```
在上述代码中,我们使用了PyTorch的transforms来对图片进行预处理。我们首先定义了一系列的transforms,包括将图片缩放到256x256,从中心裁剪出224x224的区域,将图片转换为张量,以及对张量进行归一化。然后,我们读取了一张图片,并使用上述transforms对其进行处理。最后,我们使用matplotlib库将处理后的图片显示出来。
阅读全文