如何读取 .hdf5 数据文件作为卷积神经网络的输入？

Posted 2023-03-11

【中文标题】如何读取 .hdf5 数据文件作为卷积神经网络的输入？

【英文标题】：how to read .hdf5 datafile as input to a convolutional neural network?

【发布时间】：2018-10-13 10:29:42

【问题描述】：

我已使用此链接以 HDF5 格式保存了我的大量图像及其标签： Saving and loading a large number of images (data) into a single HDF5 file 这给了我以下键。 数据集列表：

['test_img', 'test_labels', 'train_img', 'train_labels', 'train_mean', 'val_img', 'val_labels']  

现在，我想将 training data 和 training labels 提供给卷积神经网络（VGG-16 或 ResNet）用于训练目的，并且还想使用 CNN 验证和测试我的结果。如何将 HDF5 文件中的数据输入 CNN？

【问题讨论】：

搜索 load_model 和 load_weights 方法。

【参考方案1】：

看看这个开源项目tftables。

另一个thread 也有 mikkola 提供的非常详细的答案，解释了如何将您自己的生成器与新的 Tensorflow 数据集 API 一起使用。

【讨论】：

第一个链接似乎可以解决，但我还有一个问题，hdf5 表文件和 h5py 文件是否会被类似处理？我也可以对 h5py 使用相同的方法吗？

我不知道使用 h5py 和其他工具创建的 hdf5 文件之间的差异，所以我认为该方法应该适用于两者；但应该很容易快速验证。