当前位置：网站首页>pytorch中tensor转成图片保存

pytorch中tensor转成图片保存

2022-08-01 14:55:00 【passion-ma】

image_tensor为带有devices：cuda的三维tensor

image_numpy = image_tensor[0].cpu().float().numpy()  # convert it into a numpy array
image_numpy = (np.transpose(image_numpy, (1, 2, 0)) + 1) / 2.0 * 255.0  # post-processing: tranpose and scaling
image_pil = Image.fromarray(image_numpy)
image_pil.save(image_path)

原网站

版权声明
本文为[passion-ma]所创，转载请带上原文链接，感谢
https://blog.csdn.net/qq_42262610/article/details/126057918