nvs2colmap 0.3.0__tar.gz → 0.3.1__tar.gz

{nvs2colmap-0.3.0 → nvs2colmap-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvs2colmap
-Version: 0.3.0
+Version: 0.3.1
 Summary: Utilities for converting novel view synthesis datasets to COLMAP format.
 Author-email: Howard Yin <yindaheng98@gmail.com>
 Maintainer-email: Howard Yin <yindaheng98@gmail.com>
@@ -23,6 +23,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy
+Requires-Dist: plyfile
 Requires-Dist: torch
 Dynamic: license-file
 
@@ -60,7 +61,7 @@ Utilities for converting novel view synthesis datasets to COLMAP format.
 Install the Python runtime dependencies:
 
 ```bash
-pip install numpy torch
+pip install numpy plyfile torch
 ```
 
 For Neural 3D Video scenes, the command also needs `ffmpeg` and `ffprobe` on

{nvs2colmap-0.3.0 → nvs2colmap-0.3.1}/README.md

@@ -32,7 +32,7 @@ Utilities for converting novel view synthesis datasets to COLMAP format.
 Install the Python runtime dependencies:
 
 ```bash
-pip install numpy torch
+pip install numpy plyfile torch
 ```
 
 For Neural 3D Video scenes, the command also needs `ffmpeg` and `ffprobe` on

{nvs2colmap-0.3.0 → nvs2colmap-0.3.1}/nvs2colmap/dynamic3dgs/__main__.py

@@ -10,6 +10,7 @@ from nvs2colmap.write_model import write_video_colmap_text_model
 
 from .camera_meta import read_camera_meta
 from .link_frames import count_frame_dirs, link_frames
+from .points import write_init_point_cloud
 
 
 def parse_args() -> argparse.Namespace:
@@ -123,6 +124,13 @@ def main() -> None:
             use_gpu=args.colmap_use_gpu,
         )
 
+    write_init_point_cloud(
+        npz_path=folder / "init_pt_cld.npz",
+        output_pattern=frame_output_pattern / "sparse" / "0" / "points3D.ply",
+        n_frames=n_frames,
+        start_number=args.start_number,
+    )
+
     print(f"Done: {folder}")
 
 

nvs2colmap-0.3.1/nvs2colmap/dynamic3dgs/points.py

@@ -0,0 +1,25 @@
+"""Write Dynamic 3D Gaussians init_pt_cld.npz as COLMAP points3D.ply."""
+
+from __future__ import annotations
+
+from pathlib import Path
+
+import numpy as np
+
+from nvs2colmap.write_pcd import write_pcd
+
+from .link_frames import link_file
+
+
+def write_init_point_cloud(
+    npz_path: Path,
+    output_pattern: Path,
+    n_frames: int,
+    start_number: int = 1,
+) -> None:
+    data = np.load(npz_path)["data"]
+    output_pattern = str(output_pattern)
+    paths = [Path(output_pattern % frame) for frame in range(start_number, start_number + n_frames)]
+    write_pcd(paths[0], data[:, :3], data[:, 3:6])
+    for path in paths[1:]:
+        link_file(paths[0], path)

nvs2colmap-0.3.1/nvs2colmap/write_pcd.py

@@ -0,0 +1,36 @@
+"""Write xyz/rgb point clouds as PLY files."""
+
+from pathlib import Path
+
+import numpy as np
+from plyfile import PlyData, PlyElement
+
+
+def write_pcd(path, points, colors) -> None:
+    """Write a point cloud in the InstantSplat ``points3D.ply`` layout.
+
+    ``points`` and ``colors`` are ``(N, 3)`` arrays. Colors are in ``[0, 1]``
+    and are scaled to 8-bit RGB here, matching
+    ``InitializedPointCloud.save_ply``.
+    """
+    xyz = np.asarray(points)
+    rgb = np.asarray(colors) * 255.0
+    dtype = [
+        ("x", "f4"),
+        ("y", "f4"),
+        ("z", "f4"),
+        ("nx", "f4"),
+        ("ny", "f4"),
+        ("nz", "f4"),
+        ("red", "u1"),
+        ("green", "u1"),
+        ("blue", "u1"),
+    ]
+    normals = np.zeros_like(xyz)
+    elements = np.empty(xyz.shape[0], dtype=dtype)
+    attributes = np.concatenate((xyz, normals, rgb), axis=1)
+    elements[:] = list(map(tuple, attributes))
+
+    path = Path(path)
+    path.parent.mkdir(parents=True, exist_ok=True)
+    PlyData([PlyElement.describe(elements, "vertex")]).write(path)

{nvs2colmap-0.3.0 → nvs2colmap-0.3.1}/nvs2colmap.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvs2colmap
-Version: 0.3.0
+Version: 0.3.1
 Summary: Utilities for converting novel view synthesis datasets to COLMAP format.
 Author-email: Howard Yin <yindaheng98@gmail.com>
 Maintainer-email: Howard Yin <yindaheng98@gmail.com>
@@ -23,6 +23,7 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: numpy
+Requires-Dist: plyfile
 Requires-Dist: torch
 Dynamic: license-file
 
@@ -60,7 +61,7 @@ Utilities for converting novel view synthesis datasets to COLMAP format.
 Install the Python runtime dependencies:
 
 ```bash
-pip install numpy torch
+pip install numpy plyfile torch
 ```
 
 For Neural 3D Video scenes, the command also needs `ffmpeg` and `ffprobe` on

{nvs2colmap-0.3.0 → nvs2colmap-0.3.1}/nvs2colmap.egg-info/SOURCES.txt

@@ -5,9 +5,11 @@ pyproject.toml
 ./nvs2colmap/colmap.py
 ./nvs2colmap/extract_videos.py
 ./nvs2colmap/write_model.py
+./nvs2colmap/write_pcd.py
 ./nvs2colmap/dynamic3dgs/__main__.py
 ./nvs2colmap/dynamic3dgs/camera_meta.py
 ./nvs2colmap/dynamic3dgs/link_frames.py
+./nvs2colmap/dynamic3dgs/points.py
 ./nvs2colmap/n3dv/__main__.py
 ./nvs2colmap/n3dv/extract_videos.py
 ./nvs2colmap/n3dv/poses_bounds.py
@@ -20,6 +22,7 @@ nvs2colmap/__init__.py
 nvs2colmap/colmap.py
 nvs2colmap/extract_videos.py
 nvs2colmap/write_model.py
+nvs2colmap/write_pcd.py
 nvs2colmap.egg-info/PKG-INFO
 nvs2colmap.egg-info/SOURCES.txt
 nvs2colmap.egg-info/dependency_links.txt
@@ -29,6 +32,7 @@ nvs2colmap.egg-info/top_level.txt
 nvs2colmap/dynamic3dgs/__main__.py
 nvs2colmap/dynamic3dgs/camera_meta.py
 nvs2colmap/dynamic3dgs/link_frames.py
+nvs2colmap/dynamic3dgs/points.py
 nvs2colmap/n3dv/__main__.py
 nvs2colmap/n3dv/extract_videos.py
 nvs2colmap/n3dv/poses_bounds.py

{nvs2colmap-0.3.0 → nvs2colmap-0.3.1}/nvs2colmap.egg-info/requires.txt

@@ -1,2 +1,3 @@
 numpy
+plyfile
 torch

{nvs2colmap-0.3.0 → nvs2colmap-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "nvs2colmap"
-version = "0.3.0"
+version = "0.3.1"
 description = "Utilities for converting novel view synthesis datasets to COLMAP format."
 readme = "README.md"
 authors = [{ name = "Howard Yin", email = "yindaheng98@gmail.com" }]
@@ -12,6 +12,7 @@ maintainers = [{ name = "Howard Yin", email = "yindaheng98@gmail.com" }]
 requires-python = ">=3.10"
 dependencies = [
     "numpy",
+    "plyfile",
     "torch",
 ]
 keywords = [