Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-02-05 00:15:51 +01:00
Code Issues Projects Releases Wiki Activity

Change folder

Browse Source
This commit is contained in:
Nicolas Mowen 2024-12-24 07:19:49 -07:00
parent 8682b31ce5
commit 5a2113a62d
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/embeddings/maintainer.py
View File

@ -105,7 +105,7 @@ class EmbeddingMaintainer(threading.Thread):
# Lazily create the classifier. # Lazily create the classifier.
if "face_detector" not in self.__dict__: if "face_detector" not in self.__dict__:
self.__dict__["face_detector"] = cv2.FaceDetectorYN.create( self.__dict__["face_detector"] = cv2.FaceDetectorYN.create(
"/config/model_cache/facenet/facedet.onnx", "/config/model_cache/facedet/facedet.onnx",
config="", config="",
input_size=(320, 320), input_size=(320, 320),
score_threshold=0.8, score_threshold=0.8,

2
frigate/util/model.py
View File

@ -163,7 +163,7 @@ class FaceClassificationModel:
self.config = config self.config = config
self.db = db self.db = db
self.landmark_detector = cv2.face.createFacemarkLBF() self.landmark_detector = cv2.face.createFacemarkLBF()
self.landmark_detector.loadModel("/config/model_cache/facenet/landmarkdet.yaml") self.landmark_detector.loadModel("/config/model_cache/facedet/landmarkdet.yaml")
self.recognizer: cv2.face.LBPHFaceRecognizer = ( self.recognizer: cv2.face.LBPHFaceRecognizer = (
cv2.face.LBPHFaceRecognizer_create( cv2.face.LBPHFaceRecognizer_create(
radius=2, threshold=(1 - config.threshold) * 1000 radius=2, threshold=(1 - config.threshold) * 1000