pytorch的tensor，PIL和numpy的相互转换

原图：

# 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)