tensorflow版SSD使用经验

今天在服务器上搭建了tensorflow版本的ssd训练环境，其git服务器地址：
https://github.com/balancap/SSD-Tensorflow

环境

需要注意的是，使用的是python3。
原来服务器上是python2，并且tensorflow也是相应版本的，于是需要新装python3版本的。

数据

这里先使用VOC2007的数据进行测试与训练。
使用

评估

CHECKPOINT_PATH=./checkpoints/ssd_300_vgg.ckpt
DATASET_DIR=./tfrecords/voc_2007/
python3 eval_ssd_network.py \
    --eval_dir=${EVAL_DIR} \
    --dataset_dir=${DATASET_DIR} \
    --dataset_name=pascalvoc_2007 \
    --dataset_split_name=train \
    --model_name=ssd_300_vgg \
    --checkpoint_path=${CHECKPOINT_PATH} \
    --batch_size=1

进行评估。

训练

DATASET_DIR=./tfrecords/voc_2007/
TRAIN_DIR=./logs/
CHECKPOINT_PATH=./checkpoints/ssd_300_vgg.ckpt
python3 train_ssd_network.py \
    --train_dir=${TRAIN_DIR} \
    --dataset_dir=${DATASET_DIR} \
    --dataset_name=pascalvoc_2007 \
    --dataset_split_name=train \
    --model_name=ssd_300_vgg \
    --checkpoint_path=${CHECKPOINT_PATH} \
    --save_summaries_secs=60 \
    --save_interval_secs=600 \
    --weight_decay=0.0005 \
    --optimizer=adam \
    --learning_rate=0.001 \
    --batch_size=32 &