From 1dccad67d2c83209f732ff98af2d417d8121e02b Mon Sep 17 00:00:00 2001 From: Paul-Edouard Sarlin Date: Mon, 15 Jan 2024 23:44:17 +0100 Subject: [PATCH] Update example --- example.py | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/example.py b/example.py index 7cc422e..5c82c98 100644 --- a/example.py +++ b/example.py @@ -3,6 +3,8 @@ import zipfile from pathlib import Path +import enlighten + import pycolmap from pycolmap import logging @@ -28,13 +30,23 @@ def run(): if database_path.exists(): database_path.unlink() pycolmap.extract_features(database_path, image_path) - pycolmap.match_exhaustive(database_path) + num_images = pycolmap.Database(database_path).num_images if sfm_path.exists(): shutil.rmtree(sfm_path) sfm_path.mkdir(exist_ok=True) - recs = pycolmap.incremental_mapping(database_path, image_path, sfm_path) + + with enlighten.Manager() as manager: + with manager.counter(total=num_images, desc="Images registered:") as pbar: + pbar.update(0, force=True) + recs = pycolmap.incremental_mapping( + database_path, + image_path, + sfm_path, + initial_image_pair_callback=lambda: pbar.update(2), + next_image_callback=lambda: pbar.update(1), + ) for idx, rec in recs.items(): logging.info(f"#{idx} {rec.summary()}")