График замораживания из центра TensorFlow для использования в Google Coral - PullRequest
0 голосов
/ 10 мая 2019

Я пытаюсь использовать Google Colab для создания модели для использования на устройстве Google Coral.Для этого мне нужно заморозить график и затем преобразовать его в TfLite.Я могу успешно создать модель и протестировать ее с помощью учебника, но не могу понять, как заморозить график.В конце урока модель сохраняется в формате Saved_Model.Моя записная книжка Google Colab доступна здесь: https://drive.google.com/file/d/1F6wvz4yUp6Iz2hY3n7bQYnYBJe7H5JUE/view?usp=sharing

Я пробовал множество примеров кода, которые я нашел в Интернете, но все они закончились ошибками.В настоящее время я пытаюсь этот код заморозить график

from tensorflow.python.tools import freeze_graph
from tensorflow.python.saved_model import tag_constants

input_saved_model_dir = export_path
output_node_names = "dense/Softmax"
input_binary = True
input_saver_def_path = False
restore_op_name = None
filename_tensor_name = None
clear_devices = False
input_meta_graph = False
checkpoint_path = None
input_graph_filename = None
saved_model_tags = tag_constants.SERVING
output_graph_filename = "./frozenGraph.pb"

freeze_graph.freeze_graph(input_graph_filename, input_saver_def_path,
                            input_binary, checkpoint_path, output_node_names,
                              restore_op_name, filename_tensor_name,
                              output_graph_filename, clear_devices, "", "", "",
                              input_meta_graph, input_saved_model_dir,
                            saved_model_tags)

Но я получаю ошибку:

NotFoundError Traceback (most recent call last)
NotFoundError: Key module/MobilenetV2/Conv/BatchNorm/beta not found in checkpoint
[[node save_2/RestoreV2 (defined at /usr/local/lib/python3.6/dist-packages/tensorflow_hub/native_module.py:451) ]]

Также упоминается эта ошибка

During handling of the above exception, another exception occurred:

ValueError                                Traceback (most recent call last)
/usr/local/lib/python3.6/dist-packages/tensorflow/python/training/saving/saveable_object_util.py in op_list_to_dict(op_list, convert_variable_to_tensor)
    286         if name in names_to_saveables:
    287           raise ValueError("At least two variables have the same name: %s" %
--> 288                            name)
    289         names_to_saveables[name] = var
    290 

ValueError: At least two variables have the same name: dense/bias
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...