SSD: Single Shot MultiBox Detector in TensorFlow

最新推荐文章于 2022-06-03 23:00:15 发布

wxshan3

最新推荐文章于 2022-06-03 23:00:15 发布

阅读量239

点赞数

分类专栏： AI 文章标签： tensorf ssd 训练

AI 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

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'