Merge pull request #13904 from cftang0827:dnn_tf_text_graph_common_bug

pull/13931/head
Alexander Alekhin 6 years ago
commit 5fd47352cf
  1. 2
      samples/dnn/tf_text_graph_common.py

@ -324,6 +324,6 @@ def writeTextGraph(modelPath, outputPath, outNodes):
for node in graph_def.node: for node in graph_def.node:
if node.op == 'Const': if node.op == 'Const':
if 'value' in node.attr and node.attr['value'].tensor.tensor_content: if 'value' in node.attr and node.attr['value'].tensor.tensor_content:
node.attr['value'].tensor.tensor_content = '' node.attr['value'].tensor.tensor_content = b''
tf.train.write_graph(graph_def, "", outputPath, as_text=True) tf.train.write_graph(graph_def, "", outputPath, as_text=True)

Loading…
Cancel
Save