We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2f579d5 commit d8cb7f0Copy full SHA for d8cb7f0
torchvision/utils.py
@@ -90,6 +90,6 @@ def save_image(tensor, filename, nrow=8, padding=2,
90
tensor = tensor.cpu()
91
grid = make_grid(tensor, nrow=nrow, padding=padding,
92
normalize=normalize, range=range, scale_each=scale_each)
93
- 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()
94
im = Image.fromarray(ndarr)
95
im.save(filename)
0 commit comments