Skip to content

Commit

Permalink
Make it work with reid
Browse files Browse the repository at this point in the history
  • Loading branch information
Agustín Castro committed Mar 8, 2024
1 parent d5347e0 commit 93340cf
Show file tree
Hide file tree
Showing 2 changed files with 583 additions and 523 deletions.
10 changes: 8 additions & 2 deletions demos/multi_camera/src/demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -337,6 +337,12 @@ def run():
default=20,
help="Max iteration the tracked object is kept after when there are no detections",
)
parser.add_argument(
"--reid-hit-counter-max",
type=int,
default=150,
help="Maximum amount of frames trying to reidentify the object",
)
parser.add_argument(
"--nms-threshold", type=float, help="Iou threshold for detector", default=0.15
)
Expand Down Expand Up @@ -510,7 +516,7 @@ def conditional_embedding_to_spatial(detection, tracked_object):
past_detections_length=10,
reid_distance_function=embedding_distance,
reid_distance_threshold=0.5,
reid_hit_counter_max=150,
reid_hit_counter_max=args.reid_hit_counter_max,
)
tracked_objects[path] = []

Expand Down Expand Up @@ -550,7 +556,7 @@ def clusterizer_distance(tracker1, tracker2):
max_votes_split=args.max_votes_grow,
memory=args.memory,
initialization_delay=args.clusterizer_initialization_delay,
filter_by_objects_age=args.filter_by_objects_age,
reid_hit_counter_max=args.reid_hit_counter_max,
)

while True:
Expand Down
Loading

0 comments on commit 93340cf

Please sign in to comment.