将 jpg 图像转换为 .hdf5

Convert jpg images to .hdf5

在这里回答我自己的问题以供将来参考。

我最近在处理 .jpg 图像类型的数据集,需要将它们转换为 .hdf5 图像。我看到了一些反过来转换的答案,但没有从 .jpg.hdf5。最好的方法是什么?

解决方案如下所示

def convert_file(input_dir, filename, output_dir):
    filepath = input_dir + '/' + filename
    fin = open(filepath, 'rb')
    binary_data = fin.read()
    new_filepath = output_dir + '/' + filename[:-4] + '.hdf5'
    f = h5py.File(new_filepath)
    dt = h5py.special_dtype(vlen=np.dtype('uint8'))
    dset = f.create_dataset('binary_data', (100, ), dtype=dt)
    dset[0] = np.fromstring(binary_data, dtype='uint8')

我有一个工具可以做到这一点 https://github.com/raguiar2/jpg_to_h5

我还创建了自己的迷你库来处理 h5 文件。这是 link: https://github.com/ss4328/h5_manager_scripts

该脚本管理从目录创建 h5、合并 h5 以及可视化 h5 中的 12 个图像。