- home目录下创建文件夹 data/VOCdevkit。
- cd ~/data/VOCdevkit 在该目录下创建MyDataSet, 在MyDataSet下创建文件夹Annotation和JPEGImages分别存放图片文件和标注文件。
- 将get_list.py 放在MyDataSet同级目录下,运行get_list.py 生成test.txt test_name_size.txt trainval.txt三个文件。
- 进入~/caffe_ssd/data/MyDataSet, 把上面的三个文件拷贝到该目录下。
- 修改labelmap_voc.prototxt, 运行create_data.sh 生成lmdb
- lmdb存放地址:~/data/VOCdevkit/MyDataSet/lmdb
-
~/caffe_ssd/example/MyDataSet(存放数据链接,可删除)
- 进入~/work/project/MobileNet-SSD
- 运行 sh gen_model.sh 4 生成对应的模型(4是检测的类别数,根据需要填写,背景也算1类)。
- 进入example目录, 修改train.prototxt和test.prototxt.(若无特殊要求,只修改路径即可)
- cd ../
- sh train.sh
- 模型训练中。。。
-
test_draw.py
-
整个过程走完,遇到错误,随手调调即可。