保存为 HDF5 的图像未着色

Question

我目前正在开发一个将文本文件和 jpg 图像转换为 HDF5 格式的程序。用HDFView 3.0打开，图片好像只存为灰度。

hdf = h5py.File("Sample.h5")
img = Image.open("Image.jpg")
data = np.asarray((img), dtype="uint8")
hdf.create_dataset("Photos/Image 1", data=data, dtype='uint8')

dset = hdf.get("Photos/Image 1")
dset.attrs['CLASS'] = 'IMAGE'
dset.attrs['IMAGE_VERSION'] = '1.2'
arr = np.asarray([0, 255], dtype=np.uint8)
dset.attrs['IMAGE_MINMAXRANGE'] = list(arr)
dset.attrs['IMAGE_SUBCLASS'] = 'IMAGE_TRUECOLOR'
dset.attrs['INTERLACE_MODE'] = 'INTERLACE_PIXEL'

在 python 中可以使用 Image.show() 方法显示原始彩色图像：

hdf = h5py.File("Sample.h5")
array = np.array(list(hdf.get("Photos/Image 1")))
img = Image.fromarray(array.astype('uint8'))
img.show()

Answer 1

问题的第一部分。

不要问我为什么，也许 HDFview 的维护者之一可以挺身而出。要使 HDFview 能够正确显示图像，属性必须是有限长度的字符串才能被正确解释。

使用 numpy 包中的 np.string_(<string>)

import h5py  
import numpy as np
from PIL import Image

hdf = h5py.File("Sample.h5",'w')
img = Image.open("Image.jpg")
data = np.asarray((img), dtype="uint8")
hdf.create_dataset("Photos/Image 1", data=data, dtype='uint8')

dset = hdf.get("Photos/Image 1")
dset.attrs['CLASS'] = np.string_('IMAGE')
dset.attrs['IMAGE_VERSION'] = np.string_('1.2')
arr = np.asarray([0, 255], dtype=np.uint8)
dset.attrs['IMAGE_MINMAXRANGE'] = list(arr)
dset.attrs['IMAGE_SUBCLASS'] = np.string_('IMAGE_TRUECOLOR')
dset.attrs['INTERLACE_MODE'] = np.string_('INTERLACE_PIXEL')
hdf.close()

双击数据集即可在 HDFview 中显示 "Image 1"

第二个问题。

我想你正在使用 PIL 包函数 fromarray 期望 "mode of the image" 参见 https://pillow.readthedocs.io/en/3.1.x/handbook/concepts.html#concept-modes

你的情况是 RBG

因此

import h5py
import numpy as np
from PIL import Image

hdf = h5py.File("Sample.h5",'r')
array = np.array(list(hdf.get("Photos/Image 1")))
img = Image.fromarray(array.astype('uint8'), 'RGB')
img.show()

会给你

保存为 HDF5 的图像未着色

Images saved as HDF5 arent colored

python

image

hdf5

h5py

hdf