我是NVIDIA DIGITS的新手。我的训练数据集具有以下结构,其格式为.hdf5。
crops Dataset {27482, 3, 128, 192}labels Dataset {27482, 12}mean Dataset {3, 128, 192}pids Dataset {27482}
我知道如何使用更简单的格式如.txt或.jpg来喂养模型。我的问题是如何在NVIDIA DIGITS中使用.hdf5格式来喂养我的模型
回答:
HDF5数据集在DIGITS中仅用于图像分类数据集,即便如此,其支持也不够全面。
为什么呢?
Caffe对HDF5的支持远不如对LMDB的支持那么好:
由于DIGITS目前主要基于Caffe,我们的主要数据集格式是LMDB。如果/当我们支持更多后端框架时,我们可能会决定采用更通用的格式,如HDF5或zip文件。