Skip to content

Commit 33be879

Browse files
justadudewhohacksjustadudewhohacks
authored
Merge pull request justadudewhohacks#38 from justadudewhohacks/batch-input-continued
face recognition net now works with batch inputs
2 parents 8ef9b66 + 7d04da1 commit 33be879

27 files changed

+641
-182
lines changed

‎build/allFacesFactory.d.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,4 +3,4 @@ import { FaceLandmarkNet } from './faceLandmarkNet/FaceLandmarkNet';
33
import { FaceRecognitionNet } from './faceRecognitionNet/FaceRecognitionNet';
44
import { FullFaceDescription } from './FullFaceDescription';
55
import { TNetInput } from './types';
6-
export declare function allFacesFactory(detectionNet: FaceDetectionNet, landmarkNet: FaceLandmarkNet, recognitionNet: FaceRecognitionNet): (input: TNetInput, minConfidence: number) => Promise<FullFaceDescription[]>;
6+
export declare function allFacesFactory(detectionNet: FaceDetectionNet, landmarkNet: FaceLandmarkNet, recognitionNet: FaceRecognitionNet): (input: TNetInput, minConfidence: number, useBatchProcessing?: boolean) => Promise<FullFaceDescription[]>;

‎build/allFacesFactory.js

Lines changed: 30 additions & 19 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎build/allFacesFactory.js.map

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎build/faceRecognitionNet/FaceRecognitionNet.d.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ export declare class FaceRecognitionNet {
55
private _params;
66
load(weightsOrUrl: Float32Array | string | undefined): Promise<void>;
77
extractWeights(weights: Float32Array): void;
8-
forwardInput(input: NetInput): Promise<tf.Tensor2D>;
8+
forwardInput(input: NetInput): tf.Tensor2D;
99
forward(input: TNetInput): Promise<tf.Tensor2D>;
10-
computeFaceDescriptor(input: TNetInput): Promise<Float32Array>;
10+
computeFaceDescriptor(input: TNetInput): Promise<Float32Array | Float32Array[]>;
1111
}

‎build/faceRecognitionNet/FaceRecognitionNet.js

Lines changed: 40 additions & 43 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)