将图像读取为tensor，把tensor存为图像

图像读取为tensor

img = read_image("D:/PycharmProjects/Baseline_age_estimation_generic/FGNET_processed_256/079A04.JPG")
# Preprocess it for your chosen model
input_tensor = normalize(resize(img, (224, 224)) / 255., [0.485, 0.456, 0.406], [0.229, 0.224, 0.225])

def img2tensor(path):
    import torchvision.transforms as transforms
    import cv2 as cv
    img = cv.imread(path)
    transf = transforms.ToTensor()
    img_tensor = transf(img)
    # print('opencv', img)
    # print('torch', img_tensor)
    return img_tensor

tensor存为图像

def tensor2img(img,name):
    from torchvision import utils as vutils
    vutils.save_image(img, name, normalize=True)

#方式二
import torchvision.utils as vutils
with torch.no_grad():
    pred = net(x)
ts = torch.squeeze(pred.clamp(0, 1).cpu())
vutils.save_image(ts, output_dir)

from PIL import Image
im = Image.fromarray(A)
im.save("your_file.jpeg")