Проблемы с преобразованием файла FrozenGraph.pb в модель .uff для использования в Jetson Nano - PullRequest
Новая
       103

Проблемы с преобразованием файла FrozenGraph.pb в модель .uff для использования в Jetson Nano

0 голосов
/ 30 апреля 2020

Я уже некоторое время работаю с Tensorflow, создавая модели для использования с моими графическими процессорами Nvidia. Я экспериментировал с Jetson Nano, и мне нужно конвертировать некоторые из моих моделей для использования с TensorRT от Nvidia, который работает на Nano. Я искал некоторое время, пытаясь заставить это работать безуспешно. Исходя из того, что я откопал у других, это не интуитивный процесс.

У меня есть открытый вопрос с Nvidia здесь: https://forums.developer.nvidia.com/t/converting-tensorflow-pb-models-to-tensorrt-models-to-run-on-the-nano/121402, но я знаю, что это сообщество значительно больше.

Я пытался использовать tenorflow. python. tools.freeze_graph на основе предложения в этой теме: Преобразование замороженного графика Tensorflow в UFF для вывода TensorRT

Я думаю, что я близок, при работе я получаю следующую ошибку: 

python3 -m tensorflow.python.tools.freeze_graph --input_saved_model_dir /home/pal/deeppress/dp/trained_models/cardetect/saved_model/ --output_node_names test_Layers

File "/home/pal/.local/lib/python3.6/site-packages/tensorflow/python/framework/graph_util_impl.py", line 270, in convert_variables_to_constants
    inference_graph = extract_sub_graph(input_graph_def, output_node_names)

File "/home/pal/.local/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py", line 324, in new_func
    return func(*args, **kwargs)

File "/home/pal/.local/lib/python3.6/site-packages/tensorflow/python/framework/graph_util_impl.py", line 182, in extract_sub_graph
    _assert_nodes_are_present(name_to_node, dest_nodes)

File "/home/pal/.local/lib/python3.6/site-packages/tensorflow/python/framework/graph_util_impl.py", line 137, in _assert_nodes_are_present
    assert d in name_to_node, "%s is not in graph" % d
AssertionError: test_Layers is not in graph

Три вопроса: 1) Это лучший способ конвертировать мои модели TensorFlow в формат .uff? Если есть более простой способ, пожалуйста, поделитесь :) 2) Как я могу узнать, какие слои я должен указать? 3) Могу ли я просто указать testmode.uff?

Вот пример файлов в этом каталоге, я указываю на директорию сохраненной модели: введите описание изображения здесь

Заранее большое спасибо!

...