Question

我已根据提供的文档正确安装了Tensorflow对象检测API。但是，当我需要训练网络时，在research / object_detection目录中没有train.py文件。有什么我可以解决的吗？

链接：https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/installation.md

Answer 1

为澄清起见，正如Derek Chow所述，似乎火车和评估python脚本最近（约6天前）移入了“旧版”目录。假设您想继续使用旧方法。.

如果有人通过致电开始培训：

python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config

一个人会知道通过以下方式开始训练：

python legacy/train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config

Answer 2

您应该参考Running locally 教程页面上的“”部分。

这是示例配置：

#From the tensorflow/models/research/ directory
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

并运行tensorboard：

tensorboard --logdir=${MODEL_DIR}

Answer 3

您可以使用legacy训练和评估脚本，但是我们建议使用model_main。

Answer 4

在最新合并中，火车和评估移至旧目录。如果您使用的是教程，则可以使用以前的版本。

Tensorflow对象检测API没有train.py文件

4 个答案: