docs for loading 3D points

nerfstudio/data/dataparsers/nerfstudio_dataparser.py

@@ -380,6 +380,16 @@ def _generate_dataparser_outputs(self, split="train"):
         return dataparser_outputs
 
     def _load_3D_points(self, ply_file_path: Path, transform_matrix: torch.Tensor, scale_factor: float):
+        """Loads point clouds positions and colors from .ply
+
+        Args:
+            ply_file_path: Path to .ply file
+            transform_matrix: Matrix to transform world coordinates
+            scale_factor: How much to scale the camera origins by.
+
+        Returns:
+            A dictionary of points: points3D_xyz and colors: points3D_rgb
+        """
         import open3d as o3d  # Importing open3d is slow, so we only do it if we need it.
 
 

nerfstudio/process_data/colmap_utils.py

@@ -648,10 +648,8 @@ def create_ply_from_colmap(recon_dir, output_dir):
     """Writes a ply file from colmap.
 
     Args:
-        output_dir: The number of initial frames.
-
-    Returns:
-        A summary of the matching results.
+        recon_dir: Directory to grab colmap points
+        output_dir: Directory to output .ply
     """
     if (recon_dir / "points3D.bin").exists():
         colmap_points = read_points3D_binary(recon_dir / "points3D.bin")