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; });