diff --git a/torchvision/utils.py b/torchvision/utils.py index c810ce9ffa1..f07a3bb4016 100644 --- a/torchvision/utils.py +++ b/torchvision/utils.py @@ -100,6 +100,6 @@ def save_image(tensor, filename, nrow=8, padding=2, grid = make_grid(tensor, nrow=nrow, padding=padding, pad_value=pad_value, normalize=normalize, range=range, scale_each=scale_each) # Add 0.5 after unnormalizing to [0, 255] to round to nearest integer - ndarr = grid.mul_(255).add_(0.5).clamp_(0, 255).permute(1, 2, 0).to('cpu', torch.uint8).numpy() + ndarr = grid.mul(255).add_(0.5).clamp_(0, 255).permute(1, 2, 0).to('cpu', torch.uint8).numpy() im = Image.fromarray(ndarr) im.save(filename)