参考:Running Locally

1、检查数据、config文件是否配置好

可参考之前博客:

Tensorflow Object_detection之配置Training Pipeline

Tensorflow Object_detection之准备数据生成TFRecord

2、训练模型

PIPELINE_CONFIG_PATH=/data/zxx/models/research/date/VOCdevkit/faster_rcnn_resnet101_voc07.config
MODEL_DIR={path to model directory}
NUM_TRAIN_STEPS=50000
SAMPLE_1_OF_N_EVAL_EXAMPLES=1
python object_detection/model_main.py 
    --pipeline_config_path=${PIPELINE_CONFIG_PATH} 
    --num_train_steps=${NUM_TRAIN_STEPS} 
    --sample_1_of_n_eval_examples=$SAMPLE_1_OF_N_EVAL_EXAMPLES 
    --alsologtostderr 

正确姿势:

Tensorflow学习笔记5: Object_detection之训练PASCAL VOC数据集

Tensorflow学习笔记5: Object_detection之训练PASCAL VOC数据集

表示已经在训练了。

3、模型结果文件

Tensorflow学习笔记5: Object_detection之训练PASCAL VOC数据集