Commit 2f579d58 authored by TomitaNaofumi's avatar TomitaNaofumi Committed by Soumith Chintala
Browse files

clipping values, and replacing transposes with permute

Without clipping, any values larger than 255 will be replaced with
int(v mod 256) by byte(), which results in high freq noise in image.
parent 7a975284
...@@ -90,6 +90,6 @@ def save_image(tensor, filename, nrow=8, padding=2, ...@@ -90,6 +90,6 @@ def save_image(tensor, filename, nrow=8, padding=2,
tensor = tensor.cpu() tensor = tensor.cpu()
grid = make_grid(tensor, nrow=nrow, padding=padding, grid = make_grid(tensor, nrow=nrow, padding=padding,
normalize=normalize, range=range, scale_each=scale_each) normalize=normalize, range=range, scale_each=scale_each)
ndarr = grid.mul(255).byte().transpose(0, 2).transpose(0, 1).numpy() ndarr = grid.mul(255).clip(0, 255).byte().permute(1, 2, 0).numpy()
im = Image.fromarray(ndarr) im = Image.fromarray(ndarr)
im.save(filename) im.save(filename)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment