human/test/test-node.js

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 test() {
  const human = new Human(config);
  if (human) log.state('passed: create human');
  else log.error('failed: create human');

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

test();