Save instead of show?

[Lifeling]

Im looking to have the colab save the file instead of show it inline. I've got it working somewhat, but the saved image is "inverted".
Any hints on getting the image to save nicely?

[akenmorris]

Try something like this:

output = img_hr.cpu().data[0]
o = output.numpy()
o[np.where(o < 0)] = 0.0
o[np.where(o > 1)] = 1.0
output = torch.from_numpy(o)
output = T.ToPILImage()(output)
output.save(args.output)

[jwenjian]

you can also refer to my simple flask demo, but I assume @akenmorris 's code is more reasonable

https://github.com/jwenjian/artline-demo