human/demo/nodejs/node-simple.js

28 lines
1.5 KiB
JavaScript
Raw Normal View History

2021-11-18 16:10:06 +01:00
const fs = require('fs');
2022-05-22 14:50:51 +02:00
const process = require('process');
2022-01-01 14:13:04 +01:00
// eslint-disable-next-line import/no-extraneous-dependencies, no-unused-vars, @typescript-eslint/no-unused-vars
const tf = require('@tensorflow/tfjs-node'); // in nodejs environments tfjs-node is required to be loaded before human
// const faceapi = require('@vladmandic/face-api'); // use this when human is installed as module (majority of use cases)
const Human = require('../../dist/human.node.js'); // use this when using human in dev mode
2021-11-18 16:10:06 +01:00
2022-05-22 14:50:51 +02:00
const humanConfig = {
// add any custom config here
};
async function detect(inputFile) {
const human = new Human.Human(humanConfig); // create instance of human using default configuration
2021-12-27 16:59:56 +01:00
await human.load(); // optional as models would be loaded on-demand first time they are required
await human.warmup(); // optional as model warmup is performed on-demand first time its executed
2021-11-18 16:10:06 +01:00
const buffer = fs.readFileSync(inputFile); // read file data into buffer
const tensor = human.tf.node.decodeImage(buffer); // decode jpg data
2022-05-22 14:50:51 +02:00
// eslint-disable-next-line no-console
console.log('loaded input file:', inputFile, 'resolution:', tensor.shape);
2021-11-18 16:10:06 +01:00
const result = await human.detect(tensor); // run detection; will initialize backend and on-demand load models
// eslint-disable-next-line no-console
console.log(result);
}
2022-05-22 14:50:51 +02:00
if (process.argv.length === 3) detect(process.argv[2]); // if input file is provided as cmdline parameter use it
else detect('samples/in/ai-body.jpg'); // else use built-in test inputfile