用于caffe的Conver xml文件到LMDB文件

时间:2016-01-31 09:22:52

标签: python c++ xml deep-learning caffe

我想使用从训练图像数据集中提取的特征向量作为caffe卷积神经网络的输入。特征向量已保存在xml文件中。 我真的很喜欢caffe我想知道如何将xml文件转换为LMDB作为caffe输入以及如何在模型中提及我使用特征向量而不是图像。 xml文件是: enter image description here

1 个答案:

答案 0 :(得分:0)

如果你愿意使用python,那么你应该很容易。

首先,您需要解析XML文件并将数据读入一个numpy数组 您可以使用xml python package执行此操作。

第二阶段是将numpy数组中的数据写入lmdb 您可以使用this solution使用python编写LMDB。

关于caffe,只要您的输入数据有序,caffe就不关心它是否适用于图像,特征向量或任何其他输入。