From fef18d0ee037e2e6fefbd02a5582b4d53fca772e Mon Sep 17 00:00:00 2001 From: Jinwoong Date: Sun, 9 Jan 2022 19:35:48 +0900 Subject: [PATCH 1/2] Fix cmd string on tfjs export --- export.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/export.py b/export.py index 6adcf72e9e66..36c7bc62f3e4 100644 --- a/export.py +++ b/export.py @@ -346,7 +346,7 @@ def export_tfjs(keras_model, im, file, prefix=colorstr('TensorFlow.js:')): f_json = f + '/model.json' # *.json path cmd = f"tensorflowjs_converter --input_format=tf_frozen_model " \ - f"--output_node_names='Identity,Identity_1,Identity_2,Identity_3' {f_pb} {f}" + f"--output_node_names=\"Identity,Identity_1,Identity_2,Identity_3\" {f_pb} {f}" subprocess.run(cmd, shell=True) json = open(f_json).read() From 1ac02f43551bb8e77aeb868eeccbf77bc2a83a74 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sun, 9 Jan 2022 20:34:47 -1000 Subject: [PATCH 2/2] Cleanup --- export.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/export.py b/export.py index 36c7bc62f3e4..0236872c2d94 100644 --- a/export.py +++ b/export.py @@ -345,8 +345,8 @@ def export_tfjs(keras_model, im, file, prefix=colorstr('TensorFlow.js:')): f_pb = file.with_suffix('.pb') # *.pb path f_json = f + '/model.json' # *.json path - cmd = f"tensorflowjs_converter --input_format=tf_frozen_model " \ - f"--output_node_names=\"Identity,Identity_1,Identity_2,Identity_3\" {f_pb} {f}" + cmd = f'tensorflowjs_converter --input_format=tf_frozen_model ' \ + f'--output_node_names="Identity,Identity_1,Identity_2,Identity_3" {f_pb} {f}' subprocess.run(cmd, shell=True) json = open(f_json).read()