SSD: Single Shot MultiBox Detector in TensorFlow

SSD: Single Shot MultiBox Detector in TensorFlow
(1)下載tensorflow的ssd的模型代碼
(2)將./checkpoint下的模型壓縮文件解壓出來,命令爲:
    cd ./checkpoint
    unzip ssd_300_vgg.ckpt.zip
(3)製作數據集,創建bash文件,將下面的代碼放入,接着運行bash文件
    DATASET_DIR=./VOC2007/test/
    OUTPUT_DIR=./tfrecords
    python tf_convert_data.py \
    --dataset_name=pascalvoc \
    --dataset_dir=${DATASET_DIR} \
    --output_name=voc_2007_train \
    --output_dir=${OUTPUT_DIR}
(4)訓練模型:
    DATASET_DIR=./tfrecords
    TRAIN_DIR=./logs/
    CHECKPOINT_PATH=./checkpoints/ssd_300_vgg.ckpt
    python train_ssd_network.py \
    --train_dir=${TRAIN_DIR} \
    --dataset_dir=${DATASET_DIR} \
    --dataset_name=pascalvoc_2012 \
    --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
(5)測試訓練模型:
    EVAL_DIR=${TRAIN_DIR}/eval
    python eval_ssd_network.py \
    --eval_dir=${EVAL_DIR} \
    --dataset_dir=${DATASET_DIR} \
    --dataset_name=pascalvoc_2007 \
    --dataset_split_name=test \
    --model_name=ssd_300_vgg \
    --checkpoint_path=${TRAIN_DIR} \
    --wait_for_checkpoints=True \
    --batch_size=1 \
    --max_num_batches=500
(6)在tensorboard上可以看訓練的情況:
    tensorboard --logdir='/home/SSD-Tensorflow-master/log3'