Tensorflow学习笔记5: Object_detection之训练PASCAL VOC数据集
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、模型结果文件