nvs2colmap 0.3.0__tar.gz → 0.3.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvs2colmap
-Version: 0.3.0
+Version: 0.3.2
 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
@@ -124,6 +125,7 @@ For Dynamic 3D Gaussians scenes, images from both `train_meta.json` and
 `test_meta.json` are hardlinked into `frame*/images`, and included masks are
 hardlinked into `frame*/image_masks` (`cam01.jpg` pairs with `cam01.jpg.png`).
 `--no-train-camera` and `--no-test-camera` drop one split. The same
-`--use-colmap` switch writes images to `frame*/input` and runs COLMAP.
+`--use-colmap` switch writes images to `frame*/input`, masks to
+`frame*/input_mask`, and runs COLMAP.
 `--start-number` uses the same 1-based output numbering; source file
 `000000.jpg` is frame `1`.

{nvs2colmap-0.3.0 → nvs2colmap-0.3.2}/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
@@ -96,6 +96,7 @@ For Dynamic 3D Gaussians scenes, images from both `train_meta.json` and
 `test_meta.json` are hardlinked into `frame*/images`, and included masks are
 hardlinked into `frame*/image_masks` (`cam01.jpg` pairs with `cam01.jpg.png`).
 `--no-train-camera` and `--no-test-camera` drop one split. The same
-`--use-colmap` switch writes images to `frame*/input` and runs COLMAP.
+`--use-colmap` switch writes images to `frame*/input`, masks to
+`frame*/input_mask`, and runs COLMAP.
 `--start-number` uses the same 1-based output numbering; source file
 `000000.jpg` is frame `1`.

{nvs2colmap-0.3.0 → nvs2colmap-0.3.2}/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:
@@ -89,6 +90,7 @@ def main() -> None:
     n_frames = args.n_frames
     frame_output_pattern = folder / "frame%d"
     image_dir_name = "input" if args.use_colmap else "images"
+    mask_dir_name = "input_mask" if args.use_colmap else "image_masks"
     if not args.skip_frame_linking:
         if n_frames is None:
             n_frames = available_frames - args.start_number + 1
@@ -99,6 +101,7 @@ def main() -> None:
             n_frames=n_frames,
             start_number=args.start_number,
             image_dirname=image_dir_name,
+            mask_dirname=mask_dir_name,
         )
     elif n_frames is None:
         n_frames = count_frame_dirs(frame_output_pattern, start_number=args.start_number)
@@ -123,6 +126,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.0 → nvs2colmap-0.3.2}/nvs2colmap/dynamic3dgs/link_frames.py

@@ -36,6 +36,7 @@ def link_frames(
     n_frames: int,
     start_number: int = 1,
     image_dirname: str = "images",
+    mask_dirname: str = "image_masks",
 ) -> None:
     folder = Path(folder)
     frame_pattern = str(frame_pattern)
@@ -43,7 +44,7 @@ def link_frames(
         frame_index = start_number - 1 + offset
         frame_dir = Path(frame_pattern % (start_number + offset))
         image_dir = frame_dir / image_dirname
-        mask_dir = frame_dir / "image_masks"
+        mask_dir = frame_dir / mask_dirname
         for camera in cameras:
             filename = camera.filenames[frame_index]
             link_file(folder / "ims" / filename, image_dir / camera.camera.name)

nvs2colmap-0.3.2/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] * 255.0)
+    for path in paths[1:]:
+        link_file(paths[0], path)

nvs2colmap-0.3.2/nvs2colmap/write_pcd.py

@@ -0,0 +1,35 @@
+"""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 already 8-bit
+    RGB values in ``[0, 255]``.
+    """
+    xyz = np.asarray(points)
+    rgb = np.asarray(colors)
+    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.2}/nvs2colmap.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvs2colmap
-Version: 0.3.0
+Version: 0.3.2
 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
@@ -124,6 +125,7 @@ For Dynamic 3D Gaussians scenes, images from both `train_meta.json` and
 `test_meta.json` are hardlinked into `frame*/images`, and included masks are
 hardlinked into `frame*/image_masks` (`cam01.jpg` pairs with `cam01.jpg.png`).
 `--no-train-camera` and `--no-test-camera` drop one split. The same
-`--use-colmap` switch writes images to `frame*/input` and runs COLMAP.
+`--use-colmap` switch writes images to `frame*/input`, masks to
+`frame*/input_mask`, and runs COLMAP.
 `--start-number` uses the same 1-based output numbering; source file
 `000000.jpg` is frame `1`.

{nvs2colmap-0.3.0 → nvs2colmap-0.3.2}/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.2}/nvs2colmap.egg-info/requires.txt

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

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

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "nvs2colmap"
-version = "0.3.0"
+version = "0.3.2"
 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 = [