《tf object detection API使用教程--第四讲》训练VOC2012数据集

版权声明:版权归原作者所有。未经允许,严禁转载。 https://blog.csdn.net/luccao/article/details/81945780
PIPELINE_CONFIG_PATH={path to pipeline config file}
MODEL_DIR={path to model directory}
NUM_TRAIN_STEPS=50000
NUM_EVAL_STEPS=2000
python object_detection/model_main.py \
    --pipeline_config_path=${PIPELINE_CONFIG_PATH} \
    --model_dir=${MODEL_DIR} \
    --num_train_steps=${NUM_TRAIN_STEPS} \
    --num_eval_steps=${NUM_EVAL_STEPS} \
    --alsologtostderr

1. 按照自己的设置路径进行训练,参考链接

https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/running_locally.md

2. 最新版的models的代码中,已经把旧版本的train.py移到object_detection/legacy路径下,object_detection下的model_main.py代替了它的功能,不过model_main.py中没有tf.logging.set_verbosity(tf.logging.INFO)这句,导致训练的时候不打印训练信息,只要加上这句代码就行看到打印信息了(坑了我好久的)

猜你喜欢

转载自blog.csdn.net/luccao/article/details/81945780