a b/(1) PyTorch_HistoNet/util/visImage.py 

  1
  
import torch





  
  

  2
  
import numpy as np





  
  

  3
  
import matplotlib.pyplot as plt





  
  

  4
  
from util.normImageTo255 import normImageTo255





  
  

  5
  





  
  

  6
  





  
  

  7
  
def visImage(x):





  
  

  8
  
    x2 = torch.squeeze(x, 0)





  
  

  9
  
    x3 = x2.numpy()





  
  

  10
  
    x4 = np.swapaxes(x3, 0, 2)





  
  

  11
  
    x5 = normImageTo255(x4)





  
  

  12
  
    plt.imshow(x5.astype('uint8'))





  
  

  13
  
    plt.show()