Merge branch 'master' of github.com:visionml/pytracking

visionml · May 5, 2019 · bb6708d · bb6708d
2 parents ac26294 + e16d93f
commit bb6708d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/ltr/dataset/coco_seq.py b/ltr/dataset/coco_seq.py
@@ -58,8 +58,9 @@ def get_num_sequences(self):
 
     def get_sequence_info(self, seq_id):
         anno = self._get_anno(seq_id)
+        target_visible = (anno[:, 2] > 0) & (anno[:, 3] > 0)
 
-        return anno, torch.Tensor([1])
+        return anno, target_visible
 
     def _get_anno(self, seq_id):
         anno = self.coco_set.anns[self.sequence_list[seq_id]]['bbox']

diff --git a/ltr/dataset/imagenetvid.py b/ltr/dataset/imagenetvid.py
@@ -126,15 +126,14 @@ def _process_anno(self, root):
                 for tracklet_id, tracklet_start in tracklets.items():
                     tracklet_anno = []
                     target_visible = []
-                    class_name = None
+                    class_name_id = None
 
                     for f_id in range(tracklet_start, len(objects)):
                         found = False
                         for target in objects[f_id]:
                             if target.find('trackid').text == tracklet_id:
-                                if not class_name:
+                                if not class_name_id:
                                     class_name_id = target.find('name').text
-                                    class_name = self._get_class_name_from_id(class_name_id)
                                 x1 = int(target.find('bndbox/xmin').text)
                                 y1 = int(target.find('bndbox/ymin').text)
                                 x2 = int(target.find('bndbox/xmax').text)
@@ -148,7 +147,7 @@ def _process_anno(self, root):
                         if not found:
                             break
 
-                    new_sequence = {'set_id': set_id, 'vid_id': vid_id, 'class_name': class_name,
+                    new_sequence = {'set_id': set_id, 'vid_id': vid_id, 'class_name': class_name_id,
                                     'start_frame': tracklet_start, 'anno': tracklet_anno,
                                     'target_visible': target_visible, 'image_size': image_size}
                     all_sequences.append(new_sequence)