我使用此链接以 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?