diff --git a/example/node.js b/example/node.js index dfd9828..abf2a73 100644 --- a/example/node.js +++ b/example/node.js @@ -31,10 +31,10 @@ async function image(img) { const buffer = fs.readFileSync(img); const decoded = tf.node.decodeImage(buffer); const casted = decoded.toFloat(); - const image = casted.expandDims(0); + const result = casted.expandDims(0); decoded.dispose(); casted.dispose(); - return image; + return result; } async function main() { @@ -53,13 +53,11 @@ async function main() { // load face-api models log('Loading FaceAPI models'); const modelPath = path.join(__dirname, modelPathRoot); - await faceapi.nets.tinyFaceDetector.loadFromDisk(modelPath); await faceapi.nets.ssdMobilenetv1.loadFromDisk(modelPath); await faceapi.nets.ageGenderNet.loadFromDisk(modelPath); await faceapi.nets.faceLandmark68Net.loadFromDisk(modelPath); await faceapi.nets.faceRecognitionNet.loadFromDisk(modelPath); await faceapi.nets.faceExpressionNet.loadFromDisk(modelPath); - const optionsTinyFace = new faceapi.TinyFaceDetectorOptions({ inputSize: imgSize, scoreThreshold: minScore }); const optionsSSDMobileNet = new faceapi.SsdMobilenetv1Options({ minConfidence: minScore, maxResults }); // check tf engine state @@ -78,7 +76,7 @@ async function main() { .withFaceExpressions() .withFaceDescriptors() .withAgeAndGender(); - log('Image:', img, 'faces:', result.length); + log('Image:', img, 'Detected faces:', result.length); // you can access entire result object // console.log(result); tensor.dispose(); diff --git a/package.json b/package.json index 5939c07..9ac7a6c 100644 --- a/package.json +++ b/package.json @@ -10,6 +10,7 @@ "node": ">=12.0.0" }, "scripts": { + "start": "node --trace-warnings example/node.js", "build-esm": "esbuild --bundle --format=esm --target=es2018 --platform=browser --minify --sourcemap --outfile=./dist/face-api.esm.js --log-level=error --tsconfig=./tsconfig.json --external:util --external:string_decoder --external:fs src/index.ts", "build-esm-nobundle": "esbuild --bundle --format=esm --target=es2018 --platform=browser --sourcemap --outfile=./dist/face-api.esm.nobundle.js --log-level=error --tsconfig=./tsconfig.json --external:@tensorflow --external:util --external:string_decoder --external:fs --global-name=faceapi src/index.ts", "build-iife": "esbuild --bundle --format=iife --target=es2018 --platform=browser --minify --sourcemap --outfile=./dist/face-api.js --log-level=error --tsconfig=./tsconfig.json --external:util --external:string_decoder --external:fs --global-name=faceapi src/index.ts",