导入电脑图片扩增到显示扩增后的图片
时间: 2023-09-27 20:12:01 浏览: 98
要导入电脑上的图片并应用数据扩增方法后显示扩增后的图片,可以使用PIL库和torchvision库来完成。以下是一个示例代码:
```python
from PIL import Image
import torchvision.transforms as transforms
import matplotlib.pyplot as plt
# 加载图片
image = Image.open('path/to/image.jpg')
# 定义变换管道
transform = transforms.Compose([
transforms.RandomHorizontalFlip(),
transforms.RandomCrop(224),
transforms.ToTensor(), # 转换为Tensor格式
])
# 应用变换
augmented_image = transform(image)
# 将Tensor格式转为NumPy数组,并将通道维度放在最后(C, H, W -> H, W, C)
augmented_image = augmented_image.permute(1, 2, 0).numpy()
# 显示扩增后的图片
plt.imshow(augmented_image)
plt.axis('off') # 关闭坐标轴
plt.show()
```
在这个示例中,我们首先使用PIL库的`Image.open`方法加载了一张电脑上的图片。
接下来,我们定义了一个变换管道,包括随机水平翻转、随机裁剪和转换为Tensor格式。
然后,我们应用了这个变换管道,使用`transform`方法将加载的图片应用了数据扩增。
最后,我们将扩增后的图片转换为NumPy数组,并使用`plt.imshow`方法将其显示出来。
你可以根据自己的需求定义自己的变换管道,并使用PIL库和torchvision库来导入、扩增和显示电脑上的图片。
阅读全文