pytorch对图片的张量进行操作

前言

代码

import numpy as np
import torch
import torchvision
import os
from PIL import Image
from matplotlib import pyplot as plt, transforms
import torchvision.transforms as transforms

photo = Image.open('./data/hymenoptera_data/train/ants/175998972.jpg')

##PIL
img = photo.convert('RGB') # 读取图像

# transforms.ToTensor()
transform1 =transforms.Compose([
        transforms.ToTensor(),  # range [0, 255] -> [0.0,1.0]
        transforms.Scale(256),
        transforms.CenterCrop((256, 256)),
        transforms.Normalize(mean = (0.5, 0.5, 0.5), std = (0.5, 0.5, 0.5)),

    ]
)


img2 = transform1(img) # 归一化到 [0.0,1.0]
print("img2 = ",img2)
#转化为PILImage并显示

img_2 = img2/0.5+6666
print("img_2 = ",img_2)
img_2 = transforms.ToPILImage()(img2).convert('RGB')
print("img_2 = ",img_2)
img_2.show()
# img_pil_1 = np.array(photo)           # (H x W x C), [0, 255], RGB
# tensor_pl =torch.from_numpy(img_pil_1)
# # type(tensor_pl)
# img_pil_1[:,:,0]
# img_pil_1=img_pil_1/255
# print(tensor_pl)
# print(tensor_pl[0][0][0])
# print(img_pil_1)
# plt.imshow(img_pil_1)

展示