Восстановление Tensorflow 2.1 Обученная модель из Checkpoint & Meta для вывода - PullRequest
0 голосов
/ 07 апреля 2020

Я пытаюсь использовать модель, реализованную в: https://github.com/shahrukhqasim/TIES-2.0, чтобы найти строки и столбцы на изображениях в Документах. Они преобразуют аннотированные (для строк и столбцов) таблицы в tfrecords, а затем обучают CNN, используя Tensorflow. Во время Поезда конфигурации для выходного файла устанавливаются так:

summary_path=C:\projects\testing\TIES-2.0\python\summary4\
model_path=C:\projects\testing\TIES-2.0\python\mdl4\
test_out_path=C:\projects\testing\TIES-2.0\python\test4\
profiler_output_file_name=C:\projects\testing\TIES-2.0\python\profiler4
visual_feedback_out_path=C:\projects\testing\TIES-2.0\python\visual_feedback4

После того, как обучение завершено без ошибок, у меня в папке C:\projects\testing\TIES-2.0\python\mdl4\ созданы следующие файлы:

enter image description here

и отдельная папка summary4 в том же каталоге, которая имеет следующие выходные данные (обучение проводилось в три этапа):

enter image description here

Папка test4 пуста, а visual_feedback4 содержит PDF-файлы с аннотациями внутри. Используя предыдущие ответы на github (, как прогнозировать с помощью файлов .meta и контрольных точек в tenorflow? и . Используйте .ckpt и .meta -files для создания прогнозов с помощью Inception в TensorFlow ). Я пытался следующий код для восстановления модели для прогнозов:

import tensorflow.compat.v1 as tf
model_path = r"C:\projects\testing\TIES-2.0\python\mdl4"
detection_graph = tf.Graph()
with tf.Session(graph=detection_graph) as sess:
    # Load the graph with the trained states
    loader = tf.train.import_meta_graph(model_path+'\.meta')
    #loader.restore(sess, model_path)
    loader.restore(sess, r"C:\projects\testing\TIES-2.0\python\mdl4\.checkpoint.ckpt")

, который немедленно выдаёт мне ошибку:

ValueError: The passed save_path is not a valid checkpoint: C:\projects\testing\TIES-2.0\python\mdl4\.checkpoint.ckpt

попытался использовать другой файл в документе в качестве аргументов для loader.restore , так как loader сам работает без проблем. Любая помощь приветствуется, вот уже неделю.

...