将numpy.ndarray类型的矩阵保存为图片

如代码所示,res是网络生成的numpy.ndarray类型的数据,现在想要将其保存为‘.png’类型的图片。实现代码如下:

     	res = (res - res.min()) / (res.max() - res.min() + 1e-8) # res 是 numpy.ndarray 类型的[352,352]

        from PIL import Image
        import numpy as np
        image = Image.fromarray(np.uint8(res))
        image.save(opt.save_path + name)

猜你喜欢

转载自blog.csdn.net/weixin_44025103/article/details/131877798