Fix cmd string on tfjs export (ultralytics#6243)

* Fix cmd string on tfjs export * Cleanup Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
fhkiel-mlaip · Jan 10, 2022 · 8f571ee · 8f571ee
1 parent 7da7f10
commit 8f571ee
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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()