Merge pull request #13904 from cftang0827:dnn_tf_text_graph_common_bug

6 years ago · 5fd47352cf
parent dd696066b5 98d8436440
commit 5fd47352cf
1 changed files with 1 additions and 1 deletions
--- a/samples/dnn/tf_text_graph_common.py
+++ b/samples/dnn/tf_text_graph_common.py
@ -324,6 +324,6 @@ def writeTextGraph(modelPath, outputPath, outNodes):
            for node in graph_def.node:
                if node.op == 'Const':
                    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)