Caffe和LMDB - 从训练data.mdb文件生成验证集

时间:2018-04-16 01:21:57

标签: python caffe conv-neural-network lmdb

在我的数据集文件夹中,我有一个目录,我的训练数据图像转换为lmdb,如下所示:

train_lmdb/
   -- data.mdb
   -- lock.mdb

但是为了生成自定义模型,我还需要一个包含验证集的文件夹,如下所示:

val_lmdb/
   -- data.mdb
   -- lock.mdb

如何从上面的培训集文件中生成验证data.mdblock.mdb

是否可以将部分培训数据拆分为新的验证文件?

我还没有找到有关此事的任何文件,因此非常感谢SO社区的任何帮助。

1 个答案:

答案 0 :(得分:0)

在以下位置编辑imageNet示例的脚本:

<强> $ CAFFE_ROOT /示例/ imagenet / create_imagenet.sh

您将能够生成相应的培训和验证lmdb&#39。

更改train.txt和val.txt所在的路径以及匹配的图像文件的路径,并且应生成包含相应数据库的两个文件夹。