Unverified 提交 9b13a594 authored 作者: Jinwoong Yoo's avatar Jinwoong Yoo 提交者: GitHub

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

* Fix cmd string on tfjs export * Cleanup Co-authored-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com>
上级 6865d19a
...@@ -345,8 +345,8 @@ def export_tfjs(keras_model, im, file, prefix=colorstr('TensorFlow.js:')): ...@@ -345,8 +345,8 @@ def export_tfjs(keras_model, im, file, prefix=colorstr('TensorFlow.js:')):
f_pb = file.with_suffix('.pb') # *.pb path f_pb = file.with_suffix('.pb') # *.pb path
f_json = f + '/model.json' # *.json path f_json = f + '/model.json' # *.json path
cmd = f"tensorflowjs_converter --input_format=tf_frozen_model " \ cmd = f'tensorflowjs_converter --input_format=tf_frozen_model ' \
f"--output_node_names='Identity,Identity_1,Identity_2,Identity_3' {f_pb} {f}" f'--output_node_names="Identity,Identity_1,Identity_2,Identity_3" {f_pb} {f}'
subprocess.run(cmd, shell=True) subprocess.run(cmd, shell=True)
json = open(f_json).read() json = open(f_json).read()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论