我正在尝试使用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/"
我收到此错误:
警告:tensorflow:估计器的model_fn(位于0x125031b00处的.model_fn)包括params参数,但不会将params传递给Estimator。 W1025 21:53:42.973139 4590234944 model_fn.py:630]估计器的model_fn(.model_fn在0x125031b00>)包含params参数,但未将参数传递给Estimator。 INFO:tensorflow:不使用分布式协调器。 I1025 21:53:42.973598 4590234944 estimator_training.py:186]不使用分布式协调器。 INFO:tensorflow:在本地运行培训和评估(非分布式)。 I1025 21:53:42.973851 4590234944 training.py:612]在本地进行培训和评估(非分布式)。 INFO:tensorflow:开始训练并评估循环。评估将在每个检查点之后进行。检查点频率是根据RunConfig参数确定的:save_checkpoints_steps无或save_checkpoints_secs 600。 I1025 21:53:42.974123 4590234944 training.py:700]开始训练并评估循环。评估将在每个检查点之后进行。检查点频率是根据RunConfig参数确定的:save_checkpoints_steps无或save_checkpoints_secs 600。 INFO:tensorflow:跳过培训,因为max_steps已保存。 I1025 21:53:42.978157 4590234944 estimator.py:360]由于max_steps已保存,因此跳过了培训。
我该如何解决并开始训练?
答案 0 :(得分:0)
我对model_main.py
有同样的问题。
尝试从train.py
中尝试TensorFlow/models/research/object-detection/legacy/
。
我将其与以下参数一起使用:
python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_coco.config
这对我有用。