after first overfit test

2024-09-18 06:49:59 +00:00
parent d80d0ea79d
commit 0280dc7292
6 changed files with 193 additions and 42 deletions
--- a/utils/data_load.py
+++ b/utils/data_load.py
@@ -177,6 +177,9 @@ class DataLoadUtil:
            cam_to_world_R = np.asarray(label_data["extrinsic_R"])
            cam_to_world_R = DataLoadUtil.cam_pose_transformation(cam_to_world_R)
            cam_info["cam_to_world_R"] = cam_to_world_R
+            cam_to_world_O = np.asarray(label_data["extrinsic_cam_object"])
+            cam_to_world_O = DataLoadUtil.cam_pose_transformation(cam_to_world_O)
+            cam_info["cam_to_world_O"] = cam_to_world_O
        return cam_info
    
    @staticmethod