将训练有素的Tensorflow模型转换为TensorFlow.JS模型(Inception训练模型)

时间:2018-10-14 18:48:41

标签: javascript python typescript tensorflow

我正在尝试使用转移学习来训练Google的初始模型,并获得所需的输出并将其与tensaflowJS一起使用。输出应为angrysmiling

按照教程TensorFlow For Poets进行培训后,我训练了CNN,并且已经有了模型.pb

现在,我正在关注this教程,并使用了以下命令:

tensorflowjs_converter --input_format=tf_frozen_model --output_node_names='angry,smiling' --saved_model_tags=serve retrained_graph.pb web_model

但是我遇到以下错误:

  

回溯(最近通话最近):文件   “ c:\ users \ ricardo \ appdata \ local \ programs \ python \ python36 \ lib \ runpy.py”,   _run_module_as_main中的第193行       “ 主要”,mod_spec)文件“ c:\ users \ ricardo \ appdata \ local \ programs \ python \ python36 \ lib \ runpy.py”,   第85行,在_run_code中       exec(代码,run_globals)文件“ C:\ Users \ Ricardo \ AppData \ Local \ Programs \ Python \ Python36 \ Scripts \ tensorflowjs_converter.exe__main __。py”,   文件中的第9行   “ c:\ users \ ricardo \ appdata \ local \ programs \ python \ python36 \ lib \ site-packages \ tensorflowjs \ converters \ converter.py”,   272行,主要       strip_debug_ops = FLAGS.strip_debug_ops)文件“ c:\ users \ ricardo \ appdata \ local \ programs \ python \ python36 \ lib \ site-packages \ tensorflowjs \ converters \ tf_saved_model_conversion.py”,   第323行,在convert_tf_frozen_model中       图= load_graph(frozen_model_path,output_node_names)文件“ c:\ users \ ricardo \ appdata \ local \ programs \ python \ python36 \ lib \ site-packages \ tensorflowjs \ converters \ tf_saved_model_conversion.py”,   load_graph中的第67行       graph.get_operation_by_name(node.strip()))文件“ c:\ users \ ricardo \ appdata \ local \ programs \ python \ python36 \ lib \ site-packages \ tensorflow \ python \ framework \ ops.py”,   第3616行,在get_operation_by_name中       返回self.as_graph_element(name,allow_tensor = False,allow_operation = True)文件   “ c:\ users \ ricardo \ appdata \ local \ programs \ python \ python36 \ lib \ site-packages \ tensorflow \ python \ framework \ ops.py”,   第3488行,位于as_graph_element中       返回self._as_graph_element_locked(obj,allow_tensor,allow_operation)文件   “ c:\ users \ ricardo \ appdata \ local \ programs \ python \ python36 \ lib \ site-packages \ tensorflow \ python \ framework \ ops.py”,   _as_graph_element_locked中的第3548行       “图形。” %repr(name))KeyError:'名称“ \'angry”表示图中未包含的操作。'

有人知道我在做什么错吗?

0 个答案:

没有答案