生成lmdb格式数据

1.在caffe工程/data目录下本身自带有如下文件夹:

cifar10 coco ilsvrc12 ILSVRC2016 mnist VOC0712

2.在caffe工程/data目录下新建VOCdevkit文件夹(也可是自己想要的命名)

3.在VOCdevkit文件夹下建立如下目录:

caffe工程/data/VOCdevkit/VOC2007
caffe工程/data/VOCdevkit/VOC2007/Annotations
caffe工程/data/VOCdevkit/VOC2007/JPEGImages
caffe工程/data/VOCdevkit/VOC2007/ImageSets
caffe工程/data/VOCdevkit/VOC2007/ImageSets/Main

4.将caffe工程/data/VOC0712下面的create_data.sh, create_list.sh, labelmap_voc.prototxt三个文件copy到caffe工程/data/VOCdevkit/VOC2007路径下

create_list.sh: 用于生成训练集、测试集的jpg和xml路径对应关系径txt文件,及测试集目录名和对应图片大小的txt文件
create_data.sh: 用于生成lmdb格式的训练数据集和测试集
labelmap_voc.prototxt: 里头是标签的信息

5.生成训练集和测试集的txt文件,只包含文件名,不包含后缀, 放入ImageSets/Main路径下

6.修改labelmap_voc.prototxt文件为自己的类别信息

7.修改creat_list.sh文件

如下图,有四处需要修改,这里是我已经修改好的了。
其中第三处可改为自己想要的数据命名,相应的第三步的VOC2007也要与此处统一


1.png

8.修改creat_data.sh文件中涉及到路径、数据集名称的地方

9.执行creat_list.sh文件,在caffe工程/data/VOCdevkit/VOC2007路径下生成如下三个文件

train.txt test.txt test_name_size.txt

10.执行creat_data.sh文件,在caffe工程/data/VOCdevkit/VOC2007路径下生成lmdb文件夹,包含训练和测试的lmdb格式数据


参考博客:https://blog.csdn.net/jiang_ming_/article/details/81068854

注意:

  • 博客中最后一步才修改labelmap_voc.prototxt文件,这是错误的。应该在执行creat_data.sh之前就修改,因为creat_data.sh在生成lmdb数据时用到了labelmap_voc.prototx。所以避免忘记,最好在制作数据集的一开始就先修改labelmap_voc.prototxt文件

你可能感兴趣的:(生成lmdb格式数据)