admin/momentry_core
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: face tracker — add iou>0.35+dist<100 condition for same-position matching

Browse Source
This commit is contained in:
Accusys
2026-05-14 02:26:37 +08:00
parent 7e068f5bb9
commit 3a7facdc10
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/utils/face_tracker.py
View File

@@ -184,8 +184,8 @@ def match_faces(
score = 0.0 score = 0.0
# Always reject if embedding similarity is too low (different person) # Reject only if BOTH embedding AND IoU disagree (different person + different position)
if use_embedding and curr_emb and prev_emb and similarity < 0.5: if use_embedding and curr_emb and prev_emb and similarity < 0.5 and iou < 0.3:
continue continue
# Reject if bbox size changes by more than 5x (e.g., far shot → close-up) # Reject if bbox size changes by more than 5x (e.g., far shot → close-up)
@@ -209,6 +209,8 @@ def match_faces(
score = iou + similarity score = iou + similarity
elif iou > 0.5 and similarity > 0.65: elif iou > 0.5 and similarity > 0.65:
score = iou * 1.5 + similarity * 0.5 score = iou * 1.5 + similarity * 0.5
elif iou > 0.35 and distance < distance_threshold:
score = iou * 2 - distance / 500
elif similarity > 0.85: elif similarity > 0.85:
score = similarity * 2 score = similarity * 2
elif similarity > 0.75 and distance < distance_threshold: elif similarity > 0.75 and distance < distance_threshold: