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

Caffe and LMDB - generate validation set from training data.mdb files

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

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

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

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

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

是否可以将一部分训练数据拆分成新的验证文件?我还没有找到关于此事的任何文件。

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

$CAFFE_ROOT/examples/imagenet/create_imagenet.sh

您将能够生成相应的训练和验证 lmdb。

将路径更改为 train.txt 和 val.txt 所在的路径,以及它们匹配的图像文件的路径,您应该生成两个包含各自数据库的文件夹。