update for bakeup

tracking/dotrack/dotracks.py

@@ -93,14 +93,15 @@ class Track:
         self.tid = int(boxes[0, 4])
         self.cls = int(boxes[0, 6])
         self.frnum = boxes.shape[0]
-        self.imgBorder = False
+        
         self.isCornpoint = False
         self.imgshape = imgshape
-        self.state = MoveState.Unknown
+        self.isBorder = False
+        # self.state = MoveState.Unknown
         
         '''轨迹开始帧、结束帧 ID'''
-        self.start_fid = int(np.min(boxes[:, 7]))
-        self.end_fid = int(np.max(boxes[:, 7]))
+        # self.start_fid = int(np.min(boxes[:, 7]))
+        # self.end_fid = int(np.max(boxes[:, 7]))
         
         ''''''
         self.Hands = []
@@ -326,6 +327,34 @@ class Track:
         self.posState = self.Cent_isIncart+self.LB_isIncart+self.RB_isIncart
     
     
+    def is_freemove(self):
+        # if self.tid==4:
+        #     print(f"track ID: {self.tid}")
+        # boxes = self.boxes
+        # features = self.features
+        # similars = 1 - np.maximum(0.0, cdist(self.features, self.features, metric = 'cosine'))
+
+        box1 = self.boxes[0, :4]
+        box2 = self.boxes[-1, :4]
+        
+        ''' 第1帧、最后一帧subimg的相似度 '''
+        feat1 = self.features[0, :][None, :]
+        feat2 = self.features[-1, :][None, :]
+        similar = 1 - np.maximum(0.0, cdist(feat1, feat2, metric = 'cosine'))
+        condta = similar > 0.8
+
+        ''' 第1帧、最后一帧 boxes 四个角点间的距离 '''
+        ptd = box2 - box1
+        ptd1 = np.linalg.norm((ptd[0], ptd[1]))
+        ptd2 = np.linalg.norm((ptd[2], ptd[1]))
+        ptd3 = np.linalg.norm((ptd[0], ptd[3]))
+        ptd4 = np.linalg.norm((ptd[2], ptd[3]))
+        condtb = ptd1<50 and ptd2<50 and ptd3<50 and ptd4<50
+        
+        condt = condta and condtb
+        return condt
+    
+    
     def extract_hand_features(self):              
         assert self.cls == 0, "The class of traj must be HAND!"