Update v2b_dataset.py

Author: tqsdyy

V2B_main/datasets/v2b_dataset.py

@@ -243,7 +243,7 @@ class TestDataset(BaseDataset):
     def __init__(self, opts, split):
         super(TestDataset, self).__init__(
             which_dataset = opts.which_dataset,
-            path = opts.db.data_dir,
+            path = opts.db.val_data_dir,
             split = split,
             category_name = opts.db.category_name,
             offset_BB = opts.offset_BB,
@@ -354,4 +354,4 @@ def get_tracklet_lenth(self):
         return len(self.BBs)
 
     def get_PCs_and_BBs(self):
-        return self.PCs, self.BBs
+        return self.PCs, self.BBs