当我切换到模型的当前主版本(c46caa568e692a6ce5e9f388cc277b2560ea9502
)时,在尝试训练模型时出现以下错误:
WARNING:tensorflow:From /home/ubuntu/models/research/object_detection/trainer.py:210: create_global_step (from tensorflow.contrib.framework.python.ops.variables) is deprecated and will be removed in a future version.
Instructions for updating:
Please switch to tf.train.create_global_step
Traceback (most recent call last):
File "/home/ubuntu/models/research/object_detection/train.py", line 163, in <module>
tf.app.run()
File "/usr/local/lib/python3.5/dist-packages/tensorflow/python/platform/app.py", line 48, in run
_sys.exit(main(_sys.argv[:1] + flags_passthrough))
File "/home/ubuntu/models/research/object_detection/train.py", line 159, in main
worker_job_name, is_chief, FLAGS.train_dir)
File "/home/ubuntu/models/research/object_detection/trainer.py", line 217, in train
train_config.prefetch_queue_capacity, data_augmentation_options)
File "/home/ubuntu/models/research/object_detection/trainer.py", line 59, in create_input_queue
tensor_dict = create_tensor_dict_fn()
File "/home/ubuntu/models/research/object_detection/builders/input_reader_builder.py", line 72, in build
label_map_proto_file=label_map_proto_file)
TypeError: __init__() got an unexpected keyword argument 'label_map_proto_file'
如果我切换回版本b4cf230287c534bac10c78b25c4412ff209cb417
,一切都很好。不幸的是,我没有想到从哪里开始解决这个问题。