使用k-hot标签为图像数据集创建LMDB

Question

我想为图像数据集创建一个分类器，每个图像都来自所有类的多个类，因此目标值是k-hot向量。现在我创建一个文本文件，其中包含地址，如果图像文件和空间以及每行中的k-hot向量，但是当我尝试运行脚本来创建lmdb文件时，它会引发无法打开或查找文件的错误。我使用相同的数据尝试相同的过程，只使用一个数字作为类标签，一切顺利。所以我认为当标签是向量时，它无法正确解析.txt文件。

任何建议...... 谢谢

Answer 1

Caffe "Data"图层和convert_imageset脚本的编写考虑了一个非常具体的用例：图像分类。因此，caffe存储在LMDB中（并从中取出）的基本元素是Datum，其中有single integer label的空间。
_{您可以看到关于此主题here}

的更长时间的讨论

这并不意味着Caffe无法促进不同类型的输入/任务 您可以改用"HDF5Data"图层。对于hdf5输入，caffe对输入的形状和大小几乎没有限制 例如，请参阅this answer和this one，了解有关如何实际运作的详细信息。