原图:
# coding: utf-8
"""
create by ljf on 2019/3/8 15:02
"""
import torch
from torchvision import transforms
from PIL import Image
#1.PIL转为tensor
transform = transforms.ToTensor()
img = Image.open("scene.jpg")
tensor = transform(img)
# print(tensor.size())
#2.对图像增加随机翻转,随机剪切操作
transform = transforms.Compose([transforms.RandomCrop((224,224),padding=4),
transforms.RandomHorizontalFlip()])
img2 = transform(img)
# img2.show()
img2.save("scene-crop.jpg")
#3.tensor转换为numpy
arr = tensor.cpu().data.numpy()
print(arr.shape)