Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-26 19:06:11 +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 269fe43ea9
commit 8b8df6d978
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() self.__build_classifier()
cosine_index = self.labeler.transform([sub_label])[0] 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) 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 ( return (
sub_label, sub_label,
min(avg_score, svc_probability), min(avg_score, svc_probability),