tensorflow版SSD使用经验

来源:互联网 发布:淘宝女装软文推广范例 编辑:程序博客网 时间:2024/05/12 23:18

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

环境

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

数据

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

评估

CHECKPOINT_PATH=./checkpoints/ssd_300_vgg.ckptDATASET_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.ckptpython3 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 &

这里写图片描述

原创粉丝点击