zju眨眼数据集_目标检测算法SSD之训练自己的数据集

目标检测算法SSD之训练自己的数据集

prerequesties 预备知识/前提条件

下载和配置了最新SSD代码

git clone https://github.com/weiliu89/caffe ~/work/ssd

cd $_

git checkout ssd

编译caffe

下载必要的模型(包括prototxt和caffemodel)；

运行了evaluation和webcam的例子，会提示caffe的import报错。添加pycaffe路径到PYTHONPATH环境变量，或者写一个_init_paths.py来辅助引入都可以(推荐后者)。

准备自己的数据集

做成VOC2007格式的：

JPEGImages/*.png

ImageSets/Main/*.txt

Annotations/*.xml

这3个目录

生成训练用的lmdb数据

我这里数据集名叫traffic_sign，放在/home/chris/data/traffic_sign

1.复制原有脚本文件

cd ~/work/ssd

cp -R data/VOC0712 data/traffic_sign

2.修改data/traffic_sign/create_list.sh

#!/bin/bash

#root_dir=$HOME/data/VOCdevkit/

root_dir=$HOME/data/

sub_dir=ImageSets/Main

bash_dir="$(cd "$(dirname "${BASH_SOURCE[0]}")" &