用pytorch将图片变成原来的1/2。代码
时间: 2023-04-02 15:01:58 浏览: 121
segmentation:使用pytorch进行图像分割
可以使用以下代码将图片缩小为原来的一半:
```python
import torch
import torchvision.transforms as transforms
from PIL import Image
# 加载图片
img = Image.open("image.jpg")
# 定义变换
transform = transforms.Compose([
transforms.Resize((img.size[1]//2, img.size[]//2)),
transforms.ToTensor()
])
# 应用变换
img_tensor = transform(img)
# 显示图片
import matplotlib.pyplot as plt
plt.imshow(img_tensor.permute(1, 2, ))
plt.show()
```
其中,`transforms.Resize` 可以将图片大小调整为指定大小,这里将图片大小调整为原来的一半。`transforms.ToTensor` 可以将 PIL 图片转换为 PyTorch 张量。最后,使用 `matplotlib` 显示缩小后的图片。
阅读全文