参考: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
正确姿势:
表示已经在训练了。
3、模型结果文件