chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

Face recognition fixes (#15222)

Browse Source 
* Fix nginx max upload size

* Close upload dialog when done and add toasts

* Formatting

* fix ruff
This commit is contained in:
Nicolas Mowen
2024-11-27 12:48:39 -07:00
parent 5cf018ca72
commit 183e406da3
3 changed files with 39 additions and 12 deletions
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):
self.config = config
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] = {}
def __build_classifier(self) -> None:
@@ -190,11 +192,12 @@ class FaceClassificationModel:
if not self.label_map:
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:
return None
score = 1.0 - (distance / 1000)
return self.label_map[index], round(score, 2)