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也要与此处统一
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
注意:
- 博客中最后一步才修改l
abelmap_voc.prototxt
文件,这是错误的。应该在执行creat_data.sh之前就修改,因为creat_data.sh在生成lmdb数据时用到了labelmap_voc.prototx。所以避免忘记,最好在制作数据集的一开始就先修改labelmap_voc.prototxt文件