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>
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
|
349 |
-
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()
|