将数据集转换为HDF5文件

时间:2019-03-16 07:19:21

标签: deep-learning conv-neural-network hdf5 google-colaboratory

我想将面部表情数据集转换为HDF5文件,但我不知道该怎么做。

来自here,我知道如何使用cat-vs-dog数据集进行此操作,但我想知道一般情况下是如何完成的。

(所有这些目的都是为了加快Google colaboratory中的CNN模型训练过程,我尝试通过挂载google drive并因此从google drive读取我的数据集来进行此操作,但是训练模型的速度太慢(甚至比我的电脑!))。

1 个答案:

答案 0 :(得分:0)

首先,您需要导入软件包h5py

import h5py

然后,打开hdf5格式的文件,如下所示:

with h5py.File(os.path.join(output_folder, 'file_name' + '.hdf5'), 'a') as h:
// process file here as normal

有关详细信息,请参考功能create_input_files here以了解更多信息,或者您可以克隆存储库并运行python create_input_files以查看此功能的实际工作方式以及如何创建{{ 1}}个文件。