use original image for ml

2025-05-13 23:41:57 -04:00
parent c15507baad
commit 016a760dda
5 changed files with 10 additions and 77 deletions
@@ -20,7 +20,7 @@ class FaceDetector(InferenceModel):
    def _load(self) -> ModelSession:
        session = self._make_session(self.model_path)
        self.model = RetinaFace(session=session)
-        self.model.prepare(ctx_id=0, det_thresh=self.min_score, input_size=(640, 640))
+        self.model.prepare(ctx_id=0, det_thresh=self.min_score, input_size=(256, 256))

        return session