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)