Infrastructure/immich
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-01-21 00:52:43 -05:00
Code Issues Activity

fix(server): min face detection score not being passed correctly (#10050)

Browse source 
fix min score not being passed correctly
This commit is contained in:
Mert 2024-06-08 13:55:19 -04:00 • committed by GitHub
parent dbd9782763
commit df412d60c5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
server/src/interfaces/machine-learning.interface.ts
View file

@ -35,7 +35,7 @@ export type ClipTextualResponse = { [ModelTask.SEARCH]: number[] };
export type FacialRecognitionRequest = { export type FacialRecognitionRequest = {
[ModelTask.FACIAL_RECOGNITION]: { [ModelTask.FACIAL_RECOGNITION]: {
[ModelType.DETECTION]: FaceDetectionOptions; [ModelType.DETECTION]: ModelOptions & { options: { minScore: number } };
[ModelType.RECOGNITION]: ModelOptions; [ModelType.RECOGNITION]: ModelOptions;
}; };
}; };

2
server/src/repositories/machine-learning.repository.ts
View file

@ -37,7 +37,7 @@ export class MachineLearningRepository implements IMachineLearningRepository {
async detectFaces(url: string, imagePath: string, { modelName, minScore }: FaceDetectionOptions) { async detectFaces(url: string, imagePath: string, { modelName, minScore }: FaceDetectionOptions) {
const request = { const request = {
[ModelTask.FACIAL_RECOGNITION]: { [ModelTask.FACIAL_RECOGNITION]: {
[ModelType.DETECTION]: { modelName, minScore }, [ModelType.DETECTION]: { modelName, options: { minScore } },
[ModelType.RECOGNITION]: { modelName }, [ModelType.RECOGNITION]: { modelName },
}, },
}; };