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

fix: return empty descriptor for zero-sized faces

pull/34/head
Patrick Hulce 2021-01-23 12:44:14 -06:00 committed by GitHub
parent 90247697a0
commit 1e03239da4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/faceRecognitionNet/FaceRecognitionNet.ts
View File

@ -61,6 +61,10 @@ export class FaceRecognitionNet extends NeuralNetwork<NetParams> {
} }
public async computeFaceDescriptor(input: TNetInput): Promise<Float32Array|Float32Array[]> { public async computeFaceDescriptor(input: TNetInput): Promise<Float32Array|Float32Array[]> {
// When faces have a detected dimension of 0, tensorflow will crash the whole process.
// Sidestep this by returning an empty descriptor instead.
if (input.shape.some(dimension => dimension <= 0) return new Float32Array(128);
const netInput = await toNetInput(input); const netInput = await toNetInput(input);
const faceDescriptorTensors = tf.tidy( const faceDescriptorTensors = tf.tidy(