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、模型結果文件