thon | |
outputs.reconstruction.data.shape | |
torch.Size([1, 3, 880, 1072]) | |
We need to squeeze the output and get rid of axis 0, clip the values, then convert it to be numpy float. |
thon | |
outputs.reconstruction.data.shape | |
torch.Size([1, 3, 880, 1072]) | |
We need to squeeze the output and get rid of axis 0, clip the values, then convert it to be numpy float. |