Jinwoong Yoo glenn-jocher commited on
Commit
9b13a59
1 Parent(s): 6865d19

Fix `cmd` string on `tfjs` export (#6243)

Browse files

* Fix cmd string on tfjs export

* Cleanup

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Files changed (1) hide show
  1. export.py +2 -2
export.py CHANGED
@@ -345,8 +345,8 @@ def export_tfjs(keras_model, im, file, prefix=colorstr('TensorFlow.js:')):
345
  f_pb = file.with_suffix('.pb') # *.pb path
346
  f_json = f + '/model.json' # *.json path
347
 
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}"
350
  subprocess.run(cmd, shell=True)
351
 
352
  json = open(f_json).read()
 
345
  f_pb = file.with_suffix('.pb') # *.pb path
346
  f_json = f + '/model.json' # *.json path
347
 
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}'
350
  subprocess.run(cmd, shell=True)
351
 
352
  json = open(f_json).read()