numpy.array到PNG文件并返回

Question

我有一个表示灰度图像的2D numpy.array对象dtype=uint16.如何将其保存到PNG文件,然后将其读回,获取相同的数组？

Answer 1

scikit-image 让这变得非常简单：

from skimage.io import imread, imsave
import numpy as np

x = np.ones((100, 100), dtype=np.uint16)
imsave('test.png', x)
y = imread('test.png')
(x == y).all()  # True