python - 将 16 位 numpy 数组保存为 16 位 PNG 图像

Question

我正在尝试将 16 位 numpy 数组保存为 16 位 PNG，但我得到的只是一张黑色图片。我在这里放了一个我所说的最简单的例子。

im = np.random.randint(low=1, high=6536, size=65536).reshape(256,256) #sample numpy array to save as image
plt.imshow(im, cmap=plt.cm.gray)

鉴于上面的 numpy 数组，这是我用 matplotlib 看到的图像，但是当我将图像保存为 16 位 png 时，我得到了下面的图片:

import imageio

imageio.imwrite('result.png', im)

图片已保存:

可以看到一些浅灰色 Blob ，但图像基本上是黑色的。无论如何，当我读回图像并使用 matplotlib 再次对其进行可视化时，我会看到相同的起始图像。我还尝试了其他库而不是 imageio(如 PIL 或 PyPNG)，但结果相同。

我知道 16 位图像值的范围是从 0 到 65535，并且这里的数组 numpy 数组中只有从 1 到 6536 的值，但我需要保存与此类似的 numpy 数组图像，即其中表示的最大值图像不是最大可表示值。我认为保存过程中涉及某种规范化。我需要完全按照我在 matplotlib 中看到的最大分辨率保存数组，并且不压缩或缩小它们的值(因此不适合除以 255 或转换为 8 位数组)。

Answer 1

如果您在写入 PNG 文件之前将数组的数据类型转换为 numpy.uint16，看起来 imageio.imwrite 会做正确的事情:

imageio.imwrite('result.png', im.astype(np.uint16))

当我这样做时，result.png 是一个 16 位灰度 PNG 文件。

如果您希望图像具有从黑色到白色的完整灰度范围，则必须将值缩放到范围 [0, 65535]。例如。类似:

im2 = (65535*(im - im.min())/im.ptp()).astype(np.uint16)

然后你可以保存该数组

imageio.imwrite('result2.png', im2)

---

要将 NumPy 数组写入 PNG 文件，另一种方法是 numpngw (我创建的一个包)。例如，

from numpngw import write_png

im2 = (65535*(im - im.min())/im.ptp()).astype(np.uint16)
write_png('result2.png', im2)

如果您已经在使用 imageio，那么使用 numpngw 可能没有明显优势。然而，它比 imageio 的依赖要轻得多——它只依赖于 NumPy(不依赖于 PIL/Pillow，也不依赖于 libpng)。