beenull/ente
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Rename blazeface and mobilefacenet services

Browse source
This commit is contained in:
Shailesh Pandit 2022-01-06 07:22:05 +05:30
parent 19c3654926
commit 2b5aafa0bc
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/services/machineLearning/blazeFaceDetectionService.ts
View file

@ -26,7 +26,7 @@ import {
} from 'utils/machineLearning/transform';
import { enlargeBox, newBox, normFaceBox } from 'utils/machineLearning';
class TFJSFaceDetectionService implements FaceDetectionService {
class BlazeFaceDetectionService implements FaceDetectionService {
private blazeFaceModel: Promise<BlazeFaceModel>;
private blazeFaceBackModel: GraphModel;
public method: Versioned<FaceDetectionMethod>;
@ -236,4 +236,4 @@ class TFJSFaceDetectionService implements FaceDetectionService {
}
}
export default new TFJSFaceDetectionService();
export default new BlazeFaceDetectionService();

4
src/services/machineLearning/mobileFaceNetEmbeddingService.ts
View file

@ -12,7 +12,7 @@ import { imageBitmapsToTensor4D } from 'utils/machineLearning';
import { ibExtractFaceImages } from 'utils/machineLearning/faceAlign';
import { ibExtractFaceImagesFromCrops } from 'utils/machineLearning/faceCrop';
class TFJSFaceEmbeddingService implements FaceEmbeddingService {
class MobileFaceNetEmbeddingService implements FaceEmbeddingService {
private mobileFaceNetModel: Promise<tflite.TFLiteModel>;
private faceSize: number;
public method: Versioned<FaceEmbeddingMethod>;
@ -115,4 +115,4 @@ class TFJSFaceEmbeddingService implements FaceEmbeddingService {
}
}
export default new TFJSFaceEmbeddingService();
export default new MobileFaceNetEmbeddingService();