Fixed save_image to correctly output values in the full byte range. (#41)

429dbebe · Edgar Simo-Serra · Soumith Chintala · c76de96a · 429dbebe
Commit 429dbebe authored Mar 11, 2017 by Edgar Simo-Serra Committed by Soumith Chintala Mar 10, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/utils.py torchvision/utils.py +1 -1

No files found.
--- a/torchvision/utils.py
+++ b/torchvision/utils.py
@@ -50,6 +50,6 @@ def save_image(tensor, filename, nrow=8, padding=2):
    from PIL import Image
    tensor = tensor.cpu()
    grid = make_grid(tensor, nrow=nrow, padding=padding)
-    ndarr = grid.mul(0.5).add(0.5).mul(255).byte().transpose(0, 2).transpose(0, 1).numpy()
+    ndarr = grid.mul(255).byte().transpose(0,2).transpose(0,1).numpy()
    im = Image.fromarray(ndarr)
    im.save(filename)