diff --git a/torchvision/utils.py b/torchvision/utils.py index 194c6d8570a..8abf89d0075 100644 --- 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)