"...text-generation-inference.git" did not exist on "ec6d4592d59c136ea1d2cac3b4e2a7ea004a0ce3"
Commit e65925db authored by Naofumi Tomita's avatar Naofumi Tomita Committed by Soumith Chintala
Browse files

Apply clamping in save_image (#121)

* clipping values, and replacing transposes with permute

Without clipping, any values larger than 255 will be replaced with
int(v mod 256) by byte(), which results in high freq noise in image.

* replaced clip with clamp
parent 1b9304c0
...@@ -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).byte().transpose(0, 2).transpose(0, 1).numpy() ndarr = grid.mul(255).clamp(0, 255).byte().permute(1, 2, 0).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