[Fix] Fix datasets unittest (#3078)

* --fix=fix datasets unittest * --fix=remove pring
open-mmlab · Jul 12, 2024 · 168e668 · 168e668
1 parent 08bf67b
commit 168e668
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 10 deletions.
diff --git a/mmpose/datasets/datasets/wholebody3d/h3wb_dataset.py b/mmpose/datasets/datasets/wholebody3d/h3wb_dataset.py
@@ -119,6 +119,8 @@ def _load_annotations(self) -> Tuple[List[dict], List[dict]]:
 
         instance_id = 0
         for subject in self.subjects:
+            if subject not in self.ann_data:
+                continue
             actions = self.ann_data[subject].keys()
             for act in actions:
                 for cam in self.camera_order_id:

diff --git a/mmpose/datasets/datasets/wholebody3d/ubody3d_dataset.py b/mmpose/datasets/datasets/wholebody3d/ubody3d_dataset.py
@@ -84,7 +84,7 @@ def __init__(self,
 
         super().__init__(multiple_target=multiple_target, **kwargs)
 
-    METAINFO: dict = dict(from_file='configs/_base_/datasets/h3wb.py')
+    METAINFO: dict = dict(from_file='configs/_base_/datasets/ubody3d.py')
 
     def _load_ann_file(self, ann_file: str) -> dict:
         """Load annotation file."""

diff --git a/tests/data/h3wb/h3wb_train_sub.npz b/tests/data/h3wb/h3wb_train_sub.npz
diff --git a/tests/data/ubody3d/ubody3d_train.json b/tests/data/ubody3d/ubody3d_train.json
diff --git a/tests/test_datasets/test_datasets/test_wholebody_datasets/test_h3wb_dataset.py b/tests/test_datasets/test_datasets/test_wholebody_datasets/test_h3wb_dataset.py
@@ -11,7 +11,7 @@ class TestH36MWholeBodyDataset(TestCase):
     def build_h3wb_dataset(self, **kwargs):
 
         cfg = dict(
-            ann_file='h3wb_train_sub.npz',
+            ann_file='h3wb_train_bbox_subset.npz',
             data_mode='topdown',
             data_root='tests/data/h3wb',
             pipeline=[])
@@ -54,12 +54,6 @@ def test_topdown(self):
         # test topdown training
         dataset = self.build_h3wb_dataset(data_mode='topdown')
         dataset.full_init()
-        self.assertEqual(len(dataset), 3)
-        self.check_data_info_keys(dataset[0])
-
-        # test topdown testing
-        dataset = self.build_h3wb_dataset(data_mode='topdown', test_mode=True)
-        dataset.full_init()
         self.assertEqual(len(dataset), 1)
         self.check_data_info_keys(dataset[0])
 
@@ -71,5 +65,5 @@ def test_topdown(self):
             causal=False,
             pad_video_seq=True)
         dataset.full_init()
-        self.assertEqual(len(dataset), 3)
+        self.assertEqual(len(dataset), 1)
         self.check_data_info_keys(dataset[0])