mirror of https://github.com/vladmandic/human
86 lines
2.7 KiB
JavaScript
86 lines
2.7 KiB
JavaScript
const log = require('@vladmandic/pilogger');
|
|
// eslint-disable-next-line no-unused-vars, @typescript-eslint/no-unused-vars
|
|
const tf = require('@tensorflow/tfjs-node');
|
|
const Human = require('../dist/human.node.js').default;
|
|
|
|
const config = {
|
|
backend: 'tensorflow',
|
|
debug: false,
|
|
videoOptimized: false,
|
|
async: false,
|
|
warmup: 'full',
|
|
modelBasePath: 'file://models/',
|
|
filter: {
|
|
enabled: true,
|
|
},
|
|
face: {
|
|
enabled: true,
|
|
detector: { enabled: true, rotation: false },
|
|
mesh: { enabled: true },
|
|
iris: { enabled: true },
|
|
description: { enabled: true },
|
|
emotion: { enabled: true },
|
|
},
|
|
hand: {
|
|
enabled: true,
|
|
},
|
|
// body: { modelPath: 'efficientpose.json', enabled: true },
|
|
// body: { modelPath: 'blazepose.json', enabled: true },
|
|
body: { modelPath: 'posenet.json', enabled: true },
|
|
object: { enabled: true },
|
|
};
|
|
|
|
async function testInstance(human) {
|
|
if (human) log.state('passed: create human');
|
|
else log.error('failed: create human');
|
|
|
|
// if (!human.tf) human.tf = tf;
|
|
log.info('human version:', human.version);
|
|
log.info('tfjs version:', human.tf.version_core);
|
|
log.info('platform:', human.sysinfo.platform);
|
|
log.info('agent:', human.sysinfo.agent);
|
|
|
|
await human.load();
|
|
if (human.models) {
|
|
log.state('passed: load models');
|
|
const keys = Object.keys(human.models);
|
|
const loaded = keys.filter((model) => human.models[model]);
|
|
log.data(' result: defined models:', keys.length, 'loaded models:', loaded.length);
|
|
} else {
|
|
log.error('failed: load models');
|
|
}
|
|
|
|
const warmup = await human.warmup();
|
|
if (warmup) {
|
|
log.state('passed: warmup:', config.warmup);
|
|
log.data(' result: face:', warmup.face.length, 'body:', warmup.body.length, 'hand:', warmup.hand.length, 'gesture:', warmup.gesture.length, 'object:', warmup.object.length);
|
|
log.data(' result: performance:', 'load:', warmup.performance.load, 'total:', warmup.performance.total);
|
|
} else {
|
|
log.error('failed: warmup');
|
|
}
|
|
const random = tf.randomNormal([1, 1024, 1024, 3]);
|
|
const detect = await human.detect(random);
|
|
tf.dispose(random);
|
|
if (detect) {
|
|
log.state('passed: detect:', 'random');
|
|
log.data(' result: face:', detect.face.length, 'body:', detect.body.length, 'hand:', detect.hand.length, 'gesture:', detect.gesture.length, 'object:', detect.object.length);
|
|
log.data(' result: performance:', 'load:', detect.performance.load, 'total:', detect.performance.total);
|
|
} else {
|
|
log.error('failed: detect');
|
|
}
|
|
}
|
|
|
|
async function test() {
|
|
log.info('testing instance#1');
|
|
config.warmup = 'face';
|
|
const human1 = new Human(config);
|
|
await testInstance(human1);
|
|
|
|
log.info('testing instance#2');
|
|
config.warmup = 'body';
|
|
const human2 = new Human(config);
|
|
await testInstance(human2);
|
|
}
|
|
|
|
test();
|