From 22d5c18fe6a99aef16b37a863f458935cf7b3120 Mon Sep 17 00:00:00 2001
From: Jianyuan Wang <jianyuan@meta.com>
Date: Thu, 22 May 2025 23:27:58 +0100
Subject: [PATCH] [WIP] Upload demo_colmap.py, which can (a) run both vggt and
 vggt+ba, (b) output in a colmap format, (c) in the original image resolution.
 The output can be directly used for nerf/gaussian training.

---
 demo_colmap.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/demo_colmap.py b/demo_colmap.py
index b21b3f9..15f9ecf 100644
--- a/demo_colmap.py
+++ b/demo_colmap.py
@@ -228,9 +228,6 @@ def demo_fn(args):
 
         reconstruction_resolution = vggt_fixed_resolution
 
-    # Save point cloud for fast visualization
-    trimesh.PointCloud(points_3d, colors=points_rgb).export(os.path.join(args.scene_dir, "sparse/points.ply"))
-
     reconstruction = rename_colmap_recons_and_rescale_camera(
         reconstruction,
         base_image_path_list,
@@ -245,6 +242,9 @@ def demo_fn(args):
     os.makedirs(sparse_reconstruction_dir, exist_ok=True)
     reconstruction.write(sparse_reconstruction_dir)
 
+    # Save point cloud for fast visualization
+    trimesh.PointCloud(points_3d, colors=points_rgb).export(os.path.join(args.scene_dir, "sparse/points.ply"))
+
     return True