Ошибка недопустимого аргумента: загрузить (замороженную) модель Tensorflow в память (при тестировании модели на локальном компьютере) - PullRequest
Новая
       25

Ошибка недопустимого аргумента: загрузить (замороженную) модель Tensorflow в память (при тестировании модели на локальном компьютере)

0 голосов
/ 05 января 2019
  • Я использую API обнаружения объектов tenorflow.
  • Я провел обучение на удаленном сервере GPU и сохранил замороженную модель и контрольные точки.
  • После этого я взял эту замороженную модель вместе с контрольными точками и скопировал ее на свой локальный компьютер, а затем провел тестирование моих тестовых данных, используя скрипт "object_detection_tutorial.ipnyb"

Когда я запускаю эту ячейку «Загрузить (замороженную) модель Tensorflow в память», выдается ошибка invalid argument.

Не могли бы вы объяснить, в чем проблема при запуске модели сохранения на моем локальном компьютере? Необходимо ли, чтобы обучение и тестирование проводились на одной машине? Я столкнулся со следующей ошибкой:

InvalidArgumentError Traceback (самый последний вызов прошлой) ~ / Anaconda3 / Библиотека / python3.6 / сайт-пакеты / tensorflow / питон / рамки / importer.py в import_graph_def (graph_def, input_map, return_elements, name, op_dict, provider_op_list) 417 результатов = c_api.TF_GraphImportGraphDefWithResults ( -> 418 graph._c_graph, сериализованный, опции) # pylint: disable = protected-access 419 результатов = c_api_util.ScopedTFImportGraphDefResults (результаты)

InvalidArgumentError: NodeDef упоминает, что атрибут 'T' отсутствует Op selected_indices: int32>; NodeDef: {{узел Постпроцессор / BatchMultiClassNonMaxSuppression / карта / пока / MultiClassNonMaxSuppression / non_max_suppression / NonMaxSuppressionV3}} = NonMaxSuppressionV3 [T = DT_FLOAT] (Постпроцессор / BatchMultiClassNonMaxSuppression / map / while / MultiClassNonMaxSuppression / unstack, Постпроцессор / BatchMultiClassNonMaxSuppression / карта / пока / MultiClassNonMaxSuppression / перекроить, Постпроцессор / BatchMultiClassNonMaxSuppression / карта / во время / MultiClassNonMaxSuppression / Minimum, Постпроцессор / BatchMultiClassNonMaxSuppression / карта / пока / MultiClassNonMaxSuppression / non_max_suppression / iou_threshold, Постпроцессор / BatchMultiClassNonMaxSuppression / карта / пока / MultiClassNonMaxSuppression / non_max_suppression / score_threshold). (Проверьте, соответствует ли ваш двоичный файл с интерпретацией GraphDef ваш двоичный файл, генерирующий GraphDef.).

Во время обработки вышеуказанного исключения произошло другое исключение:

ValueError Traceback (самый последний вызов последний) в 5 serialized_graph = fid.read () 6 od_graph_def.ParseFromString (serialized_graph) ----> 7 tf.import_graph_def (od_graph_def, name = '')

~ / anaconda3 / Библиотека / python3.6 / сайт-пакеты / tensorflow / питон / Util / deprecation.py в new_func (* args, ** kwargs) 486 «в будущей версии», если дата «Нет» («после% s»% date), 487 инструкций) -> 488 функция возврата (* args, ** kwargs) 489 return tf_decorator.make_decorator (func, new_func, «не рекомендуется», 490 _add_deprecated_arg_notice_to_docstring (

~ / anaconda3 / Библиотека / python3.6 / сайт-пакеты / tensorflow / питон / рамки / importer.py в import_graph_def (graph_def, input_map, return_elements, name, op_dict, provider_op_list) 420 за исключением ошибок. InvalidArgumentError as e: 421 # Конвертировать в ValueError для обратной совместимости. -> 422 повысить ValueError (str (e)) 423 424 # Создать _DefinedFunctions для любых импортированных функций.

ValueError: NodeDef упоминает, что атрибут 'T' не находится в Op selected_indices: int32>; NodeDef: {{узел Постпроцессор / BatchMultiClassNonMaxSuppression / карта / пока / MultiClassNonMaxSuppression / non_max_suppression / NonMaxSuppressionV3}} = NonMaxSuppressionV3 [T = DT_FLOAT] (Постпроцессор / BatchMultiClassNonMaxSuppression / map / while / MultiClassNonMaxSuppression / unstack, Постпроцессор / BatchMultiClassNonMaxSuppression / карта / пока / MultiClassNonMaxSuppression / перекроить,Постпроцессор / BatchMultiClassNonMaxSuppression / карта / во время / MultiClassNonMaxSuppression / Minimum, Постпроцессор / BatchMultiClassNonMaxSuppression / карта / пока / MultiClassNonMaxSuppression / non_max_suppression / iou_threshold, Постпроцессор / BatchMultiClassNonMaxSuppression / карта / пока / MultiClassNonMaxSuppression / non_max_suppression / score_threshold). (Проверьте, соответствует ли ваш двоичный файл с интерпретацией GraphDef ваш двоичный файл, генерирующий GraphDef.).

...