怎么将PIL读取的一张图片,转换为tensor类型,代码尽可能简洁
时间: 2023-05-20 12:02:40 浏览: 194
基于Pil库的简易图片编辑
可以使用以下代码将PIL读取的一张图片转换为tensor类型:
```python
import torch
import torchvision.transforms as transforms
from PIL import Image
# 读取图片
img = Image.open('image.jpg')
# 定义转换器
transform = transforms.Compose([
transforms.Resize((224, 224)),
transforms.ToTensor(),
])
# 转换为tensor类型
img_tensor = transform(img)
print(img_tensor.shape) # 输出tensor的形状
```
其中,`transforms.Resize`用于将图片大小调整为224x224,`transforms.ToTensor`用于将图片转换为tensor类型。最后输出tensor的形状,以验证转换是否成功。
阅读全文