Skip to content

GitLab

Commit d8cb7f0a authored by Naofumi Tomita's avatar Naofumi Tomita Committed by Soumith Chintala
Browse files

Revert "clipping values, and replacing transposes with permute" 

This reverts commit 2f579d58.
parent 2f579d58
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/utils.py
View file @ d8cb7f0a
...@@ -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).clip(0, 255).byte().permute(1, 2, 0).numpy() ndarr = grid.mul(255).byte().transpose(0, 2).transpose(0, 1).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