首页 > 解决方案 > 如何读取 .hdf5 数据文件作为卷积神经网络的输入?

问题描述

我使用此链接以 HDF5 格式保存了我的大量图像及其标签: 将大量图像(数据)保存并加载到单个 HDF5 文件 中,该文件为我提供了以下键。
数据集列表:

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

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

标签: pythontensorflowhdf5training-dataconvolutional-neural-network

解决方案


看看这个开源项目tftables

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


推荐阅读