box select in a track and feat simi modify in tracker

tracking/trackers/bot_sort.py

@@ -172,7 +172,7 @@ class BOTSORT(BYTETracker):
             '''1. reid 相似度阈值，低于该值的两 boxes 图像不可能是同一对象，需要确定一个合理的可信阈值
                2. iou 的约束为若约束，故 iou_dists 应设置为较大的值
             '''
-            emb_dists_mask = (emb_dists > 0.9)
+            emb_dists_mask = (emb_dists > 0.8)
             iou_dists[emb_dists_mask] = 1
             emb_dists[iou_dists_mask] = 1
   

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):