Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues Projects Releases Wiki Activity

Formatting

Browse Source
This commit is contained in:
Nicolas Mowen 2024-11-19 16:46:59 -07:00
parent 2d4f5ecd6f
commit eb83c132d5
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/util/model.py
View File

@ -229,11 +229,13 @@ class FaceClassificationModel:
self.__build_classifier()
cosine_index = self.labeler.transform([sub_label])[0]
probabilities: list[float] = self.classifier.predict_proba([embedding])[0]
probabilities: np.ndarray = self.classifier.predict_proba([embedding])[0]
svc_probability = max(probabilities)
logger.debug(f"SVC face classification probability: {svc_probability} and index match: {cosine_index} / {probabilities.index(svc_probability)}")
logger.debug(
f"SVC face classification probability: {svc_probability} and index match: {cosine_index} / {np.where(probabilities == svc_probability)[0]}"
)
if cosine_index == probabilities.index(svc_probability):
if cosine_index == np.where(probabilities == svc_probability)[0]:
return (
sub_label,
min(avg_score, svc_probability),