python - caffe如何将非图像数据转换为lmdb格式

Question

我有输入大小为 [1,1,625] 的数据（宽度和高度为 1，通道数为 625）。输入大小无法更改，因为我需要使用预训练模型的权重，该模型使用此作为输入大小进行训练。我是 caffe 新手，因此无法将此数据转换为 lmdb 格式以进行 caffe 培训。每个输入单元格可以是 0 或 1，同时对于所有输入，目标值可以被认为是 1。
如果输入的大小为 [1,1,2] 并且输出中的两个值都可以是浮点数，那么如何做到这一点。

我正在尝试在我自己的数据集上运行此代码。

标签： pythoninputcaffelmdb

Answer 1

您可以将数据强制转换为 lmdb 格式，但恐怕这会很混乱。这个github 片段展示了如何使用 python 为 caffe 编写自定义 lmdb 数据。

但是，我认为在您的情况下，如果您有浮点标签，更好的策略是使用hdf5文件格式作为输入。这个答案显示了如何做到这一点。