好记性不如烂笔头,最近在使用caffe-ssd训练,所以简单写个总结,以防忘记一些过程,有不足之处,还请大家多多指教.
一.数据准备
pwd:/home/XXXXX/caffe-ssd/data/VOCdevkit/mydataset准备好jpg,txt和xml文件,分别放入对应的文件夹里
Annotations文件夹里存放所有的xml文件
ImageSets文件夹里的Main文件夹里分别存放txt文件,包括:test.txt,trainval.txt
test.txt,trainval.txt文件是自己写脚本实现的:sy_generate_txt.py
JPEGImages文件夹里存放所有的jpg文件
二.将数据转化为lmdb格式数据
pwd:/home/XXXXX/caffe-ssd/data/mydataset
在caffe-ssd根目录下执行:./data/mydataset/create_list.sh
create_list.sh文件生成训练数据的list
在caffe-ssd根目录下执行:./data/mydataset/create_data.sh
create_data.sh文件生成lmdb数据
/home/XXXXX/caffe-ssd/data/VOCdevkit/mydataset/lmdb
三.修改相关文件信息
/home/XXXXX/caffe-ssd/data/mydataset/labelmap_voc.prototxt
修改文件里面的类别信息
label:0 是背景,不需要进行任何修改 <