box select in a track and feat simi modify in tracker

2025-01-14 19:00:59 +08:00
parent 744fb7b7b2
commit bfe7bc0fd5
11 changed files with 157 additions and 22 deletions
--- a/tracking/trackers/byte_tracker.py
+++ b/tracking/trackers/byte_tracker.py
@ -374,8 +374,13 @@ class BYTETracker:
                              for x in first_finded if x.first_find]
        
        output = np.asarray(output1 + output2, dtype=np.float32)
+        
+        
+        out_feat1 = [(x.frame_id, x.idx, x.smooth_feat, x.curr_feat,  x.features) for x in self.tracked_stracks if x.is_activated]
+        out_feat2 = [(x.frame_id, x.idx, x.smooth_feat, x.curr_feat, x.features) for x in first_finded if x.first_find]
+        

-        return output
+        return output, out_feat1 + out_feat2
    
    
    def get_result(self):