nvs2colmap 0.3.1__tar.gz → 0.3.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvs2colmap
-Version: 0.3.1
+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>
@@ -125,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.1 → nvs2colmap-0.3.2}/README.md

@@ -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.1 → nvs2colmap-0.3.2}/nvs2colmap/dynamic3dgs/__main__.py

@@ -90,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
@@ -100,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)

{nvs2colmap-0.3.1 → 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.1 → nvs2colmap-0.3.2}/nvs2colmap/dynamic3dgs/points.py

@@ -20,6 +20,6 @@ def write_init_point_cloud(
     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])
+    write_pcd(paths[0], data[:, :3], data[:, 3:6] * 255.0)
     for path in paths[1:]:
         link_file(paths[0], path)

{nvs2colmap-0.3.1 → nvs2colmap-0.3.2}/nvs2colmap/write_pcd.py

@@ -9,12 +9,11 @@ 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``.
+    ``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) * 255.0
+    rgb = np.asarray(colors)
     dtype = [
         ("x", "f4"),
         ("y", "f4"),

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvs2colmap
-Version: 0.3.1
+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>
@@ -125,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.1 → nvs2colmap-0.3.2}/pyproject.toml

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