administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-03 12:07:40 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Don't return weighted name if it has the same number of results

Browse Source
This commit is contained in:
Nicolas Mowen 2025-05-24 07:22:53 -06:00
parent 6969a7fd47
commit e1b88adc35
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
frigate/data_processing/real_time/face.py
View File

@ -471,17 +471,22 @@ class FaceRealTimeProcessor(RealTimeProcessorApi):
if not results_list: if not results_list:
return None, 0.0 return None, 0.0
weighted_scores = {} counts: dict[str, int] = {}
total_weights = {} weighted_scores: dict[str, int] = {}
total_weights: dict[str, int] = {}
for name, score, face_area in results_list: for name, score, face_area in results_list:
if name == "unknown": if name == "unknown":
continue continue
if name not in weighted_scores: if name not in weighted_scores:
counts[name] = 0
weighted_scores[name] = 0.0 weighted_scores[name] = 0.0
total_weights[name] = 0.0 total_weights[name] = 0.0
# increase count
counts[name] += 1
# Capped weight based on face area # Capped weight based on face area
weight = min(face_area, max_weight) weight = min(face_area, max_weight)
@ -494,6 +499,12 @@ class FaceRealTimeProcessor(RealTimeProcessorApi):
return None, 0.0 return None, 0.0
best_name = max(weighted_scores, key=weighted_scores.get) best_name = max(weighted_scores, key=weighted_scores.get)
# If the best name has the same number of results as another name, we are not confident it is a correct result
for name, count in counts.items():
if name != best_name and counts[best_name] == count:
return None, 0.0
weighted_average = weighted_scores[best_name] / total_weights[best_name] weighted_average = weighted_scores[best_name] / total_weights[best_name]
return best_name, weighted_average return best_name, weighted_average