API-интерфейс обнаружения объектов Tensorflow ValueError: нет переменных для сохранения

Question

Я пытаюсь обучить пользовательскую модель обнаружения объекта, используя API обнаружения объекта Tenorflow.В учебных целях я использовал для сбора данных маринованные изображения, а в качестве модели использовал ssd_mobilenet_v1_coco.Когда я начал тренировку, это дало мне эту ошибку.

Traceback (most recent call last):
  File "train.py", line 184, in <module>
    tf.app.run()
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/platform/app.py", line 126, in run
    _sys.exit(main(argv))
  File "train.py", line 180, in main
    graph_hook_fn=graph_rewriter_fn)
  File "/content/models/research/object_detection/trainer.py", line 381, in train
    init_saver = tf.train.Saver(available_var_map)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 1338, in __init__
    self.build()
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 1347, in build
    self._build(self._filename, build_save=True, build_restore=True)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saver.py", line 1372, in _build
    raise ValueError("No variables to save")
ValueError: No variables to save

Полный код ошибки можно найти здесь ...

[https://gist.github.com/mpgovinda/1f59f7de7873f6ec4c4426b79dc6827a][1]

Как мне решить эту проблему?

Answer 1

Попробуйте добавить fine_tune_checkpoint_type: "detection" строку в train_config { раздел вашего *.config файла, если он не существует.

Образец:

train_config: {
  ...
  fine_tune_checkpoint: "./pre_trained_model/model.ckpt"
  fine_tune_checkpoint_type:  "detection"
  ...
}

Проверено на ssd_mobilenet_v1_quantized_300x300_coco14_syncмодель.

Answer 2

Вы должны либо использовать контрольные точки, либо удалить «init_fn = init_fn» из аргументов slim.learning.train и переменную «initializer_fn» из трейнера.Вы можете получить доступ ко всем этим из линии около 393 в тренажере

Answer 3

В последних моделях есть эта проблема, чтобы решить эту проблему:

Перейдите к файлу '.config' вашей модели и в разделе обучения измените from_detection_checkpoint: true на false

Это будет работать.

Happy Coding:)

Answer 4

У меня была такая же ошибка.Решено с помощью другой контрольной точки.

Answer 5

У меня была такая же ошибка при попытке тренироваться быстрее-rcnn с началом v2.Оказалось, что у меня была неправильная контрольная точка тонкой настройки в моем файле конфигурации модели.После того, как я связался с соответствующей контрольной точкой для модели, ошибка исчезла.