From 9b13a594e953e4e2688ff6ac0190a2247733e4ca Mon Sep 17 00:00:00 2001 From: Jinwoong Yoo Date: Mon, 10 Jan 2022 15:40:47 +0900 Subject: [PATCH] Fix `cmd` string on `tfjs` export (#6243) * Fix cmd string on tfjs export * Cleanup Co-authored-by: Glenn Jocher --- export.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/export.py b/export.py index 6adcf72e9e66..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()