小编Kaz*_*tta的帖子

我正在使用Tensorflow Object Detection API重新训练mobilenet v2 SSD,我在Windows和Ubuntu环境中都遇到了这个特殊错误.
我的环境是Windows 10,包括python 3.5,cuda 9.0和tensorflow-gpu 1.6.我用protobuf 3.4.0编译了protobuf.
我做了安装测试,没关系,所以现在我正在尝试使用自己的数据集并收到google.protobuf.text_format.ParseError: 108:7 : Message type "object_detection.protos.SsdFeatureExtractor" has no field named "batch_norm_trainable".错误.

我试过的事情:

• 使用tf 1.5 - 1.7
• 培训mobilenet v1 SSD

整个错误代码:

C:\Users\kazuy\models\research>python object_detection/train.py --logtostderr --train_dir="C:\\Users\\kazuy\\models\\research\\object_detection\\projects\\coco_selfcar\\tests\\test30" --pipeline_config_path="C:\\Users\\kazuy\\models\\research\\object_detection\\projects\\coco_selfcar\\ssd_mobilenet_v2_coco.config"
Traceback (most recent call last):
  File "object_detection/train.py", line 167, in <module>
    tf.app.run()
  File "C:\Users\kazuy\AppData\Local\Programs\Python\Python35\lib\site-packages\tensorflow\python\platform\app.py", line 126, in run
    _sys.exit(main(argv))
  File "object_detection/train.py", line 92, in main
    FLAGS.pipeline_config_path)
  File "C:\Users\kazuy\models\research\object_detection\utils\config_util.py", line 93, in get_configs_from_pipeline_file
    text_format.Merge(proto_str, pipeline_config)
  File "C:\Users\kazuy\AppData\Local\Programs\Python\Python35\lib\site-packages\google\protobuf\text_format.py", line 533, in …

我正在重新训练 TF 对象检测 API 的 mobilenet(v1)-SSD，并且在训练步骤中遇到错误。

INFO:tensorflow:Starting Session.
INFO:tensorflow:Saving checkpoint to path xxxx/model.ckpt
INFO:tensorflow:Starting Queues.
INFO:tensorflow:Error reported to Coordinator: <class 'tensorflow.python.framework.errors_impl.InvalidArgumentError'>, indices[3] = 3 is not in [0, 3)
         [[Node: cond_2/RandomCropImage/PruneCompleteleyOutsideWindow/Gather/Gather_1 = Gather[Tindices=DT_INT64, Tparams=DT_INT64, validate_indices=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](cond_2/Switch_3:1, cond_2/RandomCropImage/PruneCompleteleyOutsideWindow/Reshape)]]
INFO:tensorflow:global_step/sec: 0
INFO:tensorflow:Caught OutOfRangeError. Stopping Training.
INFO:tensorflow:Finished training! Saving model to disk.
Traceback (most recent call last):
  File "object_detection/train.py", line 168, in <module>
    tf.app.run()
  File "/home/khatta/.virtualenvs/dl/lib/python3.5/site-packages/tensorflow/python/platform/app.py", line 124, in run
    _sys.exit(main(argv))
  File "object_detection/train.py", line 165, in main
    worker_job_name, is_chief, FLAGS.train_dir)
  File "xxxx/research/object_detection/trainer.py", line …