From df412d60c532f11a117c70f07a13b18d0d43d669 Mon Sep 17 00:00:00 2001 From: Mert <101130780+mertalev@users.noreply.github.com> Date: Sat, 8 Jun 2024 13:55:19 -0400 Subject: [PATCH] fix(server): min face detection score not being passed correctly (#10050) fix min score not being passed correctly --- server/src/interfaces/machine-learning.interface.ts | 2 +- server/src/repositories/machine-learning.repository.ts | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/server/src/interfaces/machine-learning.interface.ts b/server/src/interfaces/machine-learning.interface.ts index 143281c23ac8c..5342030c8fde7 100644 --- a/server/src/interfaces/machine-learning.interface.ts +++ b/server/src/interfaces/machine-learning.interface.ts @@ -35,7 +35,7 @@ export type ClipTextualResponse = { [ModelTask.SEARCH]: number[] }; export type FacialRecognitionRequest = { [ModelTask.FACIAL_RECOGNITION]: { - [ModelType.DETECTION]: FaceDetectionOptions; + [ModelType.DETECTION]: ModelOptions & { options: { minScore: number } }; [ModelType.RECOGNITION]: ModelOptions; }; }; diff --git a/server/src/repositories/machine-learning.repository.ts b/server/src/repositories/machine-learning.repository.ts index 405e5a421d37a..b9404022efffa 100644 --- a/server/src/repositories/machine-learning.repository.ts +++ b/server/src/repositories/machine-learning.repository.ts @@ -37,7 +37,7 @@ export class MachineLearningRepository implements IMachineLearningRepository { async detectFaces(url: string, imagePath: string, { modelName, minScore }: FaceDetectionOptions) { const request = { [ModelTask.FACIAL_RECOGNITION]: { - [ModelType.DETECTION]: { modelName, minScore }, + [ModelType.DETECTION]: { modelName, options: { minScore } }, [ModelType.RECOGNITION]: { modelName }, }, };