PyTorch数据增强后的结果展示
from PIL import Image
import torch
from torchvision import transformstrans = transforms.Compose([transforms.ToTensor(), transforms.RandomErasing(p=0.9, value = 120, inplace=True)])
# 这里Compose是所做的变换img_path = '02-56-45-060-1454-camra1.bmp'
img = Image.open(img_path)
img = trans(img)image = img.cpu().clone()
image = transforms.ToPILImage()(image) # 自动转换为0-255
image.show()image.save('b.jpg')