administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-19 01:17:06 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Face recognition tuning

Browse Source
This commit is contained in:
Nicolas Mowen 2024-12-23 13:47:23 -07:00
parent a3cde4fea8
commit 0dac422d1a
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/util/model.py
View File

@ -162,8 +162,10 @@ class FaceClassificationModel:
def __init__(self, config: FaceRecognitionConfig, db: SqliteQueueDatabase): def __init__(self, config: FaceRecognitionConfig, db: SqliteQueueDatabase):
self.config = config self.config = config
self.db = db self.db = db
self.recognizer = cv2.face.LBPHFaceRecognizer_create( self.recognizer: cv2.face.LBPHFaceRecognizer = (
radius=4, threshold=(1 - config.threshold) * 1000 cv2.face.LBPHFaceRecognizer_create(
radius=2, neighbors=4, threshold=(1 - config.threshold) * 1000
)
) )
self.label_map: dict[int, str] = {} self.label_map: dict[int, str] = {}