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!"
         

tracking/dotrack/dotracks_back.py

@@ -50,6 +50,12 @@ class doBackTracks(doTracks):
         tracks = self.sub_tracks(tracks, static_tracks)
 
         
+        tracks_free = [t for t in tracks if t.frnum>1 and t.is_freemove()]
+        self.FreeMove.extend(tracks_free)
+        tracks = self.sub_tracks(tracks, tracks_free)
+        
+        
+        
         # '''购物框边界外具有运动状态的干扰目标'''
         # out_trcak = [t for t in tracks if t.is_OutTrack()]
         # tracks = self.sub_tracks(tracks, out_trcak)

tracking/dotrack/dotracks_front.py

@@ -45,6 +45,9 @@ class doFrontTracks(doTracks):
         '''剔除静止目标后的 tracks'''
         tracks = self.sub_tracks(tracks, static_tracks)
         
+        tracks_free = [t for t in tracks if t.frnum>1 and t.is_freemove()]
+        self.FreeMove.extend(tracks_free)
+        
         # [self.associate_with_hand(htrack, gtrack) for htrack in hand_tracks for gtrack in tracks]
         '''轨迹循环归并'''
         merged_tracks = self.merge_tracks_loop(tracks)

tracking/dotrack/track_back.py

@@ -217,6 +217,9 @@ class backTrack(Track):
         
         return condt
               
+
+    
+    
     def is_OutTrack(self):     
         if self.posState <= 1:
             isout = True