finish PoseDiff and NBVDataset

2024-08-30 19:21:18 +08:00
parent be5a2d57fa
commit f58360c0c0
5 changed files with 57 additions and 33 deletions
--- a/utils/data_load.py
+++ b/utils/data_load.py
@@ -129,11 +129,11 @@ class DataLoadUtil:
        target_points_camera_aug = np.concatenate([target_points_camera, np.ones((target_points_camera.shape[0], 1))], axis=-1)
        
        target_points_world = np.dot(cam_extrinsic, target_points_camera_aug.T).T[:, :3]
-        
        return {
            "points_world": target_points_world,
            "points_camera": target_points_camera
        }
+    
        
    @staticmethod
    def get_point_cloud_world_from_path(path):