store stitcher arguments in verbose mode (#227)

OpenStitching · Aug 8, 2024 · c4bde1b · c4bde1b
1 parent 4ab9bf7
commit c4bde1b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/stitching/stitcher.py b/stitching/stitcher.py
@@ -52,6 +52,7 @@ def __init__(self, **kwargs):
     def initialize_stitcher(self, **kwargs):
         self.settings = self.DEFAULT_SETTINGS.copy()
         self.validate_kwargs(kwargs)
+        self.kwargs = kwargs
         self.settings.update(kwargs)
 
         args = SimpleNamespace(**self.settings)

diff --git a/stitching/verbose.py b/stitching/verbose.py
@@ -10,6 +10,9 @@
 def verbose_stitching(stitcher, images, feature_masks=[], verbose_dir=None):
     _dir = "." if verbose_dir is None else verbose_dir
 
+    with open(verbose_output(_dir, "00_stitcher.txt"), "w") as file:
+        file.write(type(stitcher).__name__ + "(**" + str(stitcher.kwargs) + ")")
+
     images = Images.of(
         images, stitcher.medium_megapix, stitcher.low_megapix, stitcher.final_megapix
     )