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

Inline

Browse source
This commit is contained in:
Manav Rathi 2024-05-16 09:39:05 +05:30
parent 0f00803519
commit 01108141c2
No known key found for this signature in database
1 changed files with 12 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
web/apps/photos/src/services/machineLearning/machineLearningService.ts
View file

@ -96,12 +96,12 @@ const DEFAULT_ML_SYNC_CONFIG: MLSyncConfig = {
mlVersion: defaultMLVersion,
};
export const MAX_ML_SYNC_ERROR_COUNT = 1;
export const DEFAULT_ML_SEARCH_CONFIG: MLSearchConfig = {
enabled: false,
};
export const MAX_ML_SYNC_ERROR_COUNT = 1;
export async function getMLSearchConfig() {
if (isInternalUserForML()) {
return mlIDbStorage.getConfig(
@ -240,24 +240,16 @@ export class LocalMLSyncContext implements MLSyncContext {
this.config = config;
this.shouldUpdateMLVersion = shouldUpdateMLVersion;
this.faceDetectionService = MLFactory.getFaceDetectionService(
this.config.faceDetection.method,
);
this.faceCropService = MLFactory.getFaceCropService(
this.config.faceCrop.method,
);
this.faceAlignmentService = MLFactory.getFaceAlignmentService(
this.config.faceAlignment.method,
);
this.blurDetectionService = MLFactory.getBlurDetectionService(
this.config.blurDetection.method,
);
this.faceEmbeddingService = MLFactory.getFaceEmbeddingService(
this.config.faceEmbedding.method,
);
this.faceClusteringService = MLFactory.getClusteringService(
this.config.faceClustering.method,
);
this.faceDetectionService =
MLFactory.getFaceDetectionService("YoloFace");
this.faceCropService = MLFactory.getFaceCropService("ArcFace");
this.faceAlignmentService =
MLFactory.getFaceAlignmentService("ArcFace");
this.blurDetectionService =
MLFactory.getBlurDetectionService("Laplacian");
this.faceEmbeddingService =
MLFactory.getFaceEmbeddingService("MobileFaceNet");
this.faceClusteringService = MLFactory.getClusteringService("Hdbscan");
this.outOfSyncFiles = [];
this.nSyncedFiles = 0;