如何在h5格式数据中可视化图像？

Question

import h5py
f = h5py.File('the_file.h5', 'r')
one_data = f['key']
print(one_data.shape)
print(one_data.dtype)
print(one_data)

我使用上面的代码打印信息。 打印结果为：

(320, 320, 3)
uint8
<HDF5 dataset "1458552843.750": shape (320, 320, 3), type "|u1">

Answer 1

import cv2
import numpy as np
import h5py
f = h5py.File('the_file.h5', 'r')
dset = f['key']
data = np.array(dset[:,:,:])
file = 'test.jpg'
cv2.imwrite(file, data)

Answer 2

jet 提供的解决方案效果很好，但缺点是需要包含OpenCV（cv2）。如果您没有将OpenCV用于其他任何用途，则仅出于保存文件的目的安装/包含OpenCV可能有点过头。另外，您可以使用y（doc），它占用的空间更小，例如：

50.0

安装imageio就像imageio.imwrite一样简单。

此外，import imageio import numpy as np import h5py f = h5py.File('the_file.h5', 'r') dset = f['key'] data = np.array(dset[:,:,:]) file = 'test.png' # or .jpg imageio.imwrite(file, data) （doc）提供了类似的图像保存功能。