8

最適化された凍結グラフをインポートしようとすると、例外が発生します。

# read pb into graph_def
with tf.gfile.GFile(pb_file, "rb") as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())

# import graph_def
with tf.Graph().as_default() as graph:
    tf.import_graph_def(graph_def)

この行で例外を取得します。

tf.import_graph_def(graph_def)

トレースバック (最新の呼び出しが最後): ファイル
"/home/automator/PycharmProjects/tensorflow/venv/lib/python3.5/site-packages/tensorflow/python/framework/importer.py
"、行 489、
import_graph_def graph._c_graph 内、シリアル化された、オプション) # pylint: disable=protected-access tensorflow.python.framework.errors_impl.InvalidArgumentError: ノード import/final_retrain_ops/Wx_plus_b/weights_quant/AssignMinLast
の入力 0 はimport/final_retrain_ops/Wx_plus_b/weights_quant/min から float を渡されました : 0 は予想される float_ref と互換性がありません。上記の例外の処理中に、別の例外が発生しました: トレースバック (最新の呼び出しが最後):




ファイル「/snap/pycharm-community/64/helpers/pydev/pydevd.py」、1664 行目、main() ファイル「/snap/pycharm-community/64/helpers/pydev/pydevd.py」、1658 行目、メイン グローバルで
= debugger.run(setup['file'], None, None, is_module) ファイル "/snap/pycharm-community/64/helpers/pydev/pydevd.py"、1068 行目、実行中 pydev_imports.execfile( file, globals, locals) # スクリプトを実行 File
"/snap/pycharm-community/64/helpers/pydev/_pydev_imps/_pydev_execfile.py", line 18, in execfile exec(compile(contents+"\n", file, ' exec')、glob、loc) ファイル「/home/automator/PycharmProjects/tensorflow/tfliteme.py」、389 行目
、printTensors("/home/automator/Desktop/cervix/optimized_model.pb")
ファイル "/home/automator/PycharmProjects/tensorflow/tfliteme.py"、374 行目
、printTensors tf.import_graph_def(graph_def) ファイル "
/home/automator/PycharmProjects/tensorflow/venv/lib/python3.5/site-packages/ tensorflow/python/util/deprecation.py", line 432, in
new_func return func(*args, **kwargs) File "
/home/automator/PycharmProjects/tensorflow/venv/lib/python3.5/site-packages/tensorflow /python/framework/importer.py」、493行目、
import_graph_def raise ValueError(str(e)) ValueError: ノード import/final_retrain_ops/Wx_plus_b/weights_quant/AssignMinLast の入力 0 は import/final_retrain_ops/Wx_plus_b/weights_quant/ から float で 渡さ
れました
min:0 と互換性
がありません

float_ref が必要です。

4

1 に答える 1

2

あなたpb_fileが正しい形式（このimport_graph_def()ようなもの）であることを確認し、次のように「インポート」のデフォルト値を上書きしようとするために、「名前」パラメータに値を設定してみてください。

# read pb into graph_def
with tf.gfile.GFile(pb_file, "rb") as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())

# import graph_def
with tf.Graph().as_default() as graph:
    tf.import_graph_def(graph_def, name='')
于 2019-01-21T11:42:47.220 に答える