Unverified 提交 489920ab authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

TF.js fix (#9536)

* TF.js fix May resolve https://github.com/ultralytics/yolov5/issues/9534Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update ci-testing.yml Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> * Update ci-testing.yml Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> * Update ci-testing.yml Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> Signed-off-by: 's avatarGlenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: 's avatarpre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
上级 999482b4
...@@ -485,7 +485,7 @@ class TFModel: ...@@ -485,7 +485,7 @@ class TFModel:
iou_thres, iou_thres,
conf_thres, conf_thres,
clip_boxes=False) clip_boxes=False)
return nms, x[1] return (nms,)
return x # output [1,6300,85] = [xywh, conf, class0, class1, ...] return x # output [1,6300,85] = [xywh, conf, class0, class1, ...]
# x = x[0] # [x(1,6300,85), ...] to x(6300,85) # x = x[0] # [x(1,6300,85), ...] to x(6300,85)
# xywh = x[..., :4] # x(6300,4) boxes # xywh = x[..., :4] # x(6300,4) boxes
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论