diff --git a/src/faceRecognitionNet/FaceRecognitionNet.ts b/src/faceRecognitionNet/FaceRecognitionNet.ts index 297c200..25c6f61 100644 --- a/src/faceRecognitionNet/FaceRecognitionNet.ts +++ b/src/faceRecognitionNet/FaceRecognitionNet.ts @@ -61,6 +61,10 @@ export class FaceRecognitionNet extends NeuralNetwork { } public async computeFaceDescriptor(input: TNetInput): Promise { + // 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 faceDescriptorTensors = tf.tidy(