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
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/92798188