add fps

2024-10-06 11:49:03 +08:00
parent e315fd99ee
commit a84417ef62
2 changed files with 30 additions and 20 deletions
--- a/core/nbv_dataset.py
+++ b/core/nbv_dataset.py
@@ -162,8 +162,8 @@ class NBVReconstructionDataset(BaseDataset):
        )

        combined_scanned_views_pts = np.concatenate(scanned_views_pts, axis=0)
-        voxel_downsampled_combined_scanned_pts_np = (
-            PtsUtil.voxel_downsample_point_cloud(combined_scanned_views_pts, 0.002)
+        voxel_downsampled_combined_scanned_pts_np, _ = (
+            PtsUtil.voxelize_points(combined_scanned_views_pts, 0.002)
        )
        random_downsampled_combined_scanned_pts_np = (
            PtsUtil.random_downsample_point_cloud(