Function main

python/examples/custom_incremental_pipeline.py:462–504

(
    database_path: Path,
    image_path: Path,
    output_path: Path,
    options: IncrementalPipelineOptions | None = None,
    input_path: Path | None = None,
)

def main(
    database_path: Path,
    image_path: Path,
    output_path: Path,
    options: IncrementalPipelineOptions | None = None,
    input_path: Path | None = None,
) -> dict[int, Reconstruction]:
    if options is None:
        options = IncrementalPipelineOptions()
    options.image_path = image_path
    if not database_path.exists():
        logging.fatal(f"Database path does not exist: {database_path}")
    if not image_path.exists():
        logging.fatal(f"Image path does not exist: {image_path}")
    output_path.mkdir(exist_ok=True, parents=True)
    reconstruction_manager = ReconstructionManager()
    if input_path:
        reconstruction_manager.read(input_path)

    with pycolmap.Database.open(database_path) as database:
        mapper = IncrementalPipeline(options, database, reconstruction_manager)
        num_images = database.num_images()
        with enlighten.Manager() as manager:
            with manager.counter(
                total=num_images, desc="Images registered:"
            ) as pbar:
                pbar.update(0, force=True)
                mapper.add_callback(
                    IncrementalPipelineCallback.INITIAL_IMAGE_PAIR_REG_CALLBACK,
                    lambda: pbar.update(2),
                )
                mapper.add_callback(
                    IncrementalPipelineCallback.NEXT_IMAGE_REG_CALLBACK,
                    lambda: pbar.update(1),
                )
                main_incremental_mapper(mapper)

    # write and output
    reconstruction_manager.write(output_path)
    reconstructions = {}
    for i in range(reconstruction_manager.size()):
        reconstructions[i] = reconstruction_manager.get(i)
    return reconstructions


def parse_args() -> argparse.Namespace: