add target_pts_num into dataset

2024-09-29 18:11:55 +08:00
parent cb983fdc74
commit 99e57c3f4c
2 changed files with 144 additions and 114 deletions
--- a/utils/data_load.py
+++ b/utils/data_load.py
@@ -98,6 +98,13 @@ class DataLoadUtil:
            scene_info = json.load(f)
        return scene_info
    
+    @staticmethod
+    def load_target_pts_num_dict(root, scene_name):
+        target_pts_num_path = os.path.join(root, scene_name, "target_pts_num.json")
+        with open(target_pts_num_path, "r") as f:
+            target_pts_num_dict = json.load(f)
+        return target_pts_num_dict
+    
    @staticmethod
    def load_target_object_pose(root, scene_name):
        scene_info = DataLoadUtil.load_scene_info(root, scene_name)