balancap/SSD-Tensorflow训练自己的数据集

1、首先clone一下：git clone https://github.com/balancap/SSD-Tensorflow.git

2、按照这个博客的详解进行一步一步走：https://blog.csdn.net/w5688414/article/details/78395177

3、注意参考：https://blog.csdn.net/Echo_Harrington/article/details/81131441

1. 修改datasets/pascalvoc_common.py中24行的labels，除了背景不改，其他的改成自己的labels
2. datasets/pascalvoc_to_tfrecords.py文件中第82行读取图片格式改为相对应的‘.jpg’或‘.jpeg’，83行‘r’改为‘rb’
3. nets/ssd_vgg_300.py文件中第96-97行的num_classes和no_annotation_label改为“类别数+1”,比如5类则改为6
4. eval_ssd_network.py 文件中第66行修改num_classes为“类别数+1”
5. train_ssd_network.py文件中第修改135行的num_classes为“类别数+1”
6. 在训练时的checkpoint_path应该这样输入：CHECKPOINT_PATH=./checkpoints/ssd_300_vgg.ckpt/ssd_300_vgg.ckpt，否则会出现save_path出错
7. 多gpus训练：https://github.com/balancap/SSD-Tensorflow/issues/58