Skip to content

Commit 8f571ee

Browse files
Jinwoong Yooglenn-jocher
andauthored
Fix cmd string on tfjs export (ultralytics#6243)
* Fix cmd string on tfjs export * Cleanup Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
1 parent 7da7f10 commit 8f571ee

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

export.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -345,8 +345,8 @@ def export_tfjs(keras_model, im, file, prefix=colorstr('TensorFlow.js:')):
345345
f_pb = file.with_suffix('.pb') # *.pb path
346346
f_json = f + '/model.json' # *.json path
347347

348-
cmd = f"tensorflowjs_converter --input_format=tf_frozen_model " \
349-
f"--output_node_names='Identity,Identity_1,Identity_2,Identity_3' {f_pb} {f}"
348+
cmd = f'tensorflowjs_converter --input_format=tf_frozen_model ' \
349+
f'--output_node_names="Identity,Identity_1,Identity_2,Identity_3" {f_pb} {f}'
350350
subprocess.run(cmd, shell=True)
351351

352352
json = open(f_json).read()

0 commit comments

Comments
 (0)