Log In Sign Up
Models:
Wanda Brown
WandaB/
HistoTNet
0
Downloads: 1
[ad8447]: / (2) PyTorch_HistoTNet / util / visImage.py

Download this file

14 lines (11 with data), 288 Bytes 

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
import torch

import numpy as np

import matplotlib.pyplot as plt

from util.normImageTo255 import normImageTo255





def visImage(x):

    x2 = torch.squeeze(x, 0)

    x3 = x2.numpy()

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

    x5 = normImageTo255(x4)

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

    plt.show()