Ошибка при экспорте объекта Tf при экспорте графа - PullRequest
Новая
       25

Ошибка при экспорте объекта Tf при экспорте графа

0 голосов
/ 10 октября 2019

У меня проблема с экспортом графа с использованием API обнаружения объектов tenorflow.

Я запускаю следующий код: 

!python export_inference_graph.py --input_type=image_tensor --pipeline_config_path=training/ssd_mobilenet_v2_coco.config --trained_checkpoint_prefix=training/model.ckpt-1001 --output_directory=output

И вот ошибка: 

Traceback (most recent call last):
  File "export_inference_graph.py", line 162, in <module>
    tf.app.run()
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/platform/app.py", line 40, in run
    _run(main=main, argv=argv, flags_parser=_parse_flags_tolerate_undef)
  File "/usr/local/lib/python3.6/dist-packages/absl/app.py", line 299, in run
    _run_main(main, args)
  File "/usr/local/lib/python3.6/dist-packages/absl/app.py", line 250, in _run_main
    sys.exit(main(argv))
  File "export_inference_graph.py", line 158, in main
    write_inference_graph=FLAGS.write_inference_graph)
  File "/content/drive/My Drive/OD/models-master/research/object_detection/exporter.py", line 489, in export_inference_graph
    write_inference_graph=write_inference_graph)
  File "/content/drive/My Drive/OD/models-master/research/object_detection/exporter.py", line 418, in _export_inference_graph
    trained_checkpoint_prefix=checkpoint_to_use)
  File "/content/drive/My Drive/OD/models-master/research/object_detection/exporter.py", line 327, in write_graph_and_checkpoint
    saver.restore(sess, trained_checkpoint_prefix)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/training/saver.py", line 1282, in restore
    checkpoint_prefix)
ValueError: The passed save_path is not a valid checkpoint: training/model.ckpt-1001

У меня проблема, хотя у меня model.ckpt-1001в папке

...