NVIDIA DIGITS中的HDF5输入数据集使用情况

Question

我是使用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格式提供我的模型

Answer 1

HDF5数据集仅用于DIGITS中的图像分类数据集，即使这样，支持也不是非常全功能。

为什么？

Caffe不支持HDF5，它支持LMDB：

• 对于大型数据集，您必须将它们分解为单独的文件（see here）
• 未预取数据 - 整个数据集一次性读入内存（see here）
• HDF5Data图层（see here）
不支持数据转换

由于DIGITS目前主要基于Caffe，因此我们的主要数据集格式是LMDB。如果/当我们支持更多的后端框架时，我们可能会决定采用更通用的格式，如HDF5或zipfiles。