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

Formatting

Browse Source
This commit is contained in:
Nicolas Mowen 2024-11-27 12:38:44 -07:00
parent ce1ee0a28f
commit bcd27a80ee
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
frigate/util/model.py
View File

@ -162,7 +162,9 @@ 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(radius=4, threshold=(1 - config.threshold) * 1000) self.recognizer = cv2.face.LBPHFaceRecognizer_create(
radius=4, threshold=(1 - config.threshold) * 1000
)
self.label_map: dict[int, str] = {} self.label_map: dict[int, str] = {}
def __build_classifier(self) -> None: def __build_classifier(self) -> None:
@ -190,11 +192,12 @@ class FaceClassificationModel:
if not self.label_map: if not self.label_map:
self.__build_classifier() self.__build_classifier()
index, distance = self.recognizer.predict(cv2.equalizeHist(cv2.cvtColor(face_image, cv2.COLOR_BGR2GRAY))) index, distance = self.recognizer.predict(
cv2.equalizeHist(cv2.cvtColor(face_image, cv2.COLOR_BGR2GRAY))
)
if index == -1: if index == -1:
return None return None
score = 1.0 - (distance / 1000) score = 1.0 - (distance / 1000)
return self.label_map[index], round(score, 2) return self.label_map[index], round(score, 2)