Keras 将TensorFlow hub中的图形冻结在Google Coral上使用

Keras 将TensorFlow hub中的图形冻结在Google Coral上使用,keras,tensorflow-hub,google-coral,tensorflow-lite,Keras,Tensorflow Hub,Google Coral,Tensorflow Lite,我正在尝试使用GoogleColab来构建一个在GoogleCoral设备上使用的模型。为了做到这一点,我需要冻结图形,然后将其转换为TfLite。我可以使用教程成功地创建模型并对其进行测试,但我不知道如何冻结图形。在本教程结束时,模型将以保存的模型格式保存。我的Google Colab笔记本可在以下位置获得: 我尝试了在网上找到的各种代码示例,但都以错误告终。目前我正在尝试用这段代码冻结图形 从tensorflow.python.tools导入冻结图 从tensorflow.python.sa

我正在尝试使用GoogleColab来构建一个在GoogleCoral设备上使用的模型。为了做到这一点,我需要冻结图形,然后将其转换为TfLite。我可以使用教程成功地创建模型并对其进行测试,但我不知道如何冻结图形。在本教程结束时,模型将以保存的模型格式保存。我的Google Colab笔记本可在以下位置获得:

我尝试了在网上找到的各种代码示例,但都以错误告终。目前我正在尝试用这段代码冻结图形

从tensorflow.python.tools导入冻结图
从tensorflow.python.saved_model导入标记_常量
输入\保存\模型\目录=导出\路径
输出节点名称=“密集/Softmax”
输入\二进制=真
输入_saver_def_path=False
恢复\操作\名称=无
filename\u tensor\u name=None
清除设备=错误
输入元图=假
检查点路径=无
输入\图形\文件名=无
已保存的\u模型\u标记=标记\u常量
output_graph_filename=“./frozenGraph.pb”
冻结图形。冻结图形(输入图形文件名、输入保存文件路径、,
输入\二进制、检查点\路径、输出\节点\名称,
还原\u op\u名称、文件名\u张量\u名称,
输出\u图形\u文件名,清除\u设备,“”,“”,“”,
输入元图,输入保存的模型目录,
已保存的\u模型\u标记)
但我得到了一个错误:

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
你能仔细看看你是否发现了有用的东西吗?你能仔细看看你是否发现了有用的东西吗?