python - 培训不起作用。 "Skipping training since max_steps has already saved"

Question

我正在尝试使用 TensorFlow 训练对象检测器。我下载了预训练的模型 ssd_mobilenet_v1_coco_2018_01_28，创建了我的 .pbtxt 文件、我的训练和测试记录。

我使用的是 TensorFlow/models/research/object-detection 中的 model_main.py

当我尝试开始训练时"python model_main.py --logtostderr model_dir=results/--pipeline_config_path=training/ssd_mobilenet_v1_pets.config --model_dir=ssd_mobilenet_v1_coco_2018_01_28/"

我收到此错误:

WARNING:tensorflow:Estimator's model_fn (.model_fn at 0x125031b00>) includes params argument, but params are not passed to Estimator. W1025 21:53:42.973139 4590234944 model_fn.py:630] Estimator's model_fn (.model_fn at 0x125031b00>) includes params argument, but params are not passed to Estimator. INFO:tensorflow:Not using Distribute Coordinator. I1025 21:53:42.973598 4590234944 estimator_training.py:186] Not using Distribute Coordinator. INFO:tensorflow:Running training and evaluation locally (non-distributed). I1025 21:53:42.973851 4590234944 training.py:612] Running training and evaluation locally (non-distributed). INFO:tensorflow:Start train and evaluate loop. The evaluate will happen after every checkpoint. Checkpoint frequency is determined based on RunConfig arguments: save_checkpoints_steps None or save_checkpoints_secs 600. I1025 21:53:42.974123 4590234944 training.py:700] Start train and evaluate loop. The evaluate will happen after every checkpoint. Checkpoint frequency is determined based on RunConfig arguments: save_checkpoints_steps None or save_checkpoints_secs 600. INFO:tensorflow:Skipping training since max_steps has already saved. I1025 21:53:42.978157 4590234944 estimator.py:360] Skipping training since max_steps has already saved.

我该如何解决这个问题并开始训练？

Answer 1

我在 model_main.py 中遇到了同样的问题。

尝试使用 TensorFlow/models/research/object-detection/legacy/ 中的 train.py。

我将它与以下参数一起使用:

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

这对我有用。