pytorch 归一化与反归一化

 

ToTensor中就有转到0-1之间了。

 

# -*- coding:utf-8 -*-


import time

import torch

from torchvision import transforms

import cv2

transform_val_list = [
    # transforms.Resize(size=(160, 160), interpolation=3),  # Image.BICUBIC
    transforms.ToTensor(),
    transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])
]

trans_compose = transforms.Compose(transform_val_list)



if __name__ == '__main__':
    std= [0.229, 0.224, 0.225]
    mean=[0.485, 0.456, 0.406]
    path="d:/2.jpg"

    data=cv2.imread(path)
    t1 = time.time()
    x = trans_compose(data)
    x[0]=x[0]*std[0]+mean[0]
    x[1]=x[1]*std[1]+mean[1]
    x[2]=x[2].mul(std[2])+mean[2]

    img = x.mul(255).byte()
    img = img.numpy().transpose((1, 2, 0))
    # torch.set_num_threads(3)
    # img=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
    cv2.imshow("sdf", img)
    cv2.w