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

web worker fixes

pull/293/head
Vladimir Mandic 2020-11-15 09:28:57 -05:00
parent 92b4ac71dd
commit a4da0b7fe2
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
demo/browser.js
View File

@ -247,7 +247,7 @@ function webWorker(input, image, canvas, timestamp) {
}
// pass image data as arraybuffer to worker by reference to avoid copy
if (ui.bench) bench.begin();
worker.postMessage({ image: image.data.buffer, width: canvas.width, height: canvas.height }, [image.data.buffer]);
worker.postMessage({ image: image.data.buffer, width: canvas.width, height: canvas.height, userConfig }, [image.data.buffer]);
}
// main processing function when input is webcam, can use direct invocation or web worker
@ -272,7 +272,10 @@ function runHumanDetect(input, canvas, timestamp) {
status('');
if (ui.useWorker) {
// get image data from video as we cannot send html objects to webworker
const offscreen = new OffscreenCanvas(canvas.width, canvas.height);
const offscreen = (typeof OffscreenCanvas !== 'undefined') ? new OffscreenCanvas(canvas.width, canvas.height) : document.createElement('canvas');
offscreen.width = canvas.width;
offscreen.height = canvas.height;
const ctx = offscreen.getContext('2d');
ctx.drawImage(input, 0, 0, input.width, input.height, 0, 0, canvas.width, canvas.height);
const data = ctx.getImageData(0, 0, canvas.width, canvas.height);
@ -475,12 +478,12 @@ async function main() {
document.getElementById('log').innerText = `Human: version ${human.version} TensorFlow/JS: version ${human.tf.version_core}`;
// human.tf.ENV.set('WEBGL_FORCE_F16_TEXTURES', true);
// this is not required, just pre-loads all models
if (ui.modelsPreload) {
if (ui.modelsPreload && !ui.useWorker) {
status('loading');
await human.load(userConfig);
}
// this is not required, just pre-warms all models for faster initial inference
if (ui.modelsWarmup) {
if (ui.modelsWarmup && !ui.useWorker) {
status('initializing');
sample = await human.warmup(userConfig, document.getElementById('sample-image'));
}

2
src/human.js
View File

@ -182,7 +182,7 @@ class Human {
/* debug mode is really too mcuh
tf.enableDebugMode();
*/
if (this.config.backend === 'webgl') {
if (tf.getBackend() === 'webgl') {
if (this.config.deallocate) {
this.log('changing webgl: WEBGL_DELETE_TEXTURE_THRESHOLD:', this.config.deallocate);
tf.ENV.set('WEBGL_DELETE_TEXTURE_THRESHOLD', this.config.deallocate ? 0 : -1);