ambiser
/
human
mirror of https://github.com/vladmandic/human
9
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix posenet

pull/356/head
Vladimir Mandic 2021-12-18 12:24:01 -05:00
parent 44ad8c6d4d
commit 4d8feaff3e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/body/posenet.ts
View File

@ -164,7 +164,7 @@ export async function predict(input: Tensor, config: Config): Promise<BodyResult
const normalized = tf.sub(tf.div(tf.cast(resized, 'float32'), 127.5), 1.0); const normalized = tf.sub(tf.div(tf.cast(resized, 'float32'), 127.5), 1.0);
const results: Array<Tensor> = model.execute(normalized, poseNetOutputs) as Array<Tensor>; const results: Array<Tensor> = model.execute(normalized, poseNetOutputs) as Array<Tensor>;
const results3d = results.map((y) => tf.squeeze(y, [0])); 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; return results3d;
}); });