modified for site test

tracking/utils/proBoxes.py

@@ -12,10 +12,12 @@ class Boxes:
         """Initialize the Boxes class."""
         if boxes.ndim == 1:
             boxes = boxes[None, :]
-        n = boxes.shape[-1]
-        assert n in (6, 7, 8), f'expected `n` in [6, 7], but got {n}'  # xyxyb, track_id, conf, cls
+        m, n = boxes.shape
+        assert n in (6, 7), f'expected `n` in [6, 7], but got {n}'  # xyxy, track_id, conf, cls
+        
+        '''对每一个box进行编号，利用该编号可以索引对应 feature'''
+        self.data = np.concatenate([boxes[:, :4], np.arange(m).reshape(-1, 1), boxes[:, 4:]], axis=-1)
         
-        self.data = boxes
         self.orig_shape = orig_shape
     
     def cpu(self):
@@ -30,10 +32,9 @@ class Boxes:
         """Return the boxes in xyxy format."""
         return self.data[:, :4]
     
-    
     @property
     def xyxyb(self):
-        """Return the boxes in xyxyb format."""
+        """Return the boxes in xyxyb format."""    
         return self.data[:, :5]
 
     @property