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


参考: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、模型结果文件

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM