From 4d8feaff3e4ab6213989bb92e55787605a6afba5 Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sat, 18 Dec 2021 12:24:01 -0500 Subject: [PATCH] fix posenet --- src/body/posenet.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/body/posenet.ts b/src/body/posenet.ts index 724aa69c..1b30a042 100644 --- a/src/body/posenet.ts +++ b/src/body/posenet.ts @@ -164,7 +164,7 @@ export async function predict(input: Tensor, config: Config): Promise = model.execute(normalized, poseNetOutputs) as Array; const results3d = results.map((y) => tf.squeeze(y, [0])); - results3d[1] = results3d[1].sigmoid(); // apply sigmoid on scores + results3d[1] = tf.sigmoid(results3d[1]); // apply sigmoid on scores return results3d; });