Improve affine stitcher (#20)

Author: lukasalexanderweber

README.md

@@ -1,6 +1,6 @@
 # stitching
 
-A Python package for fast and robust Image Stitching. 
+A Python package for fast and robust Image Stitching.
 
 Based on opencv's [stitching
 module](https://github.com/opencv/opencv/tree/4.x/modules/stitching)

stitching/cli/stitch.py

@@ -53,7 +53,7 @@
     action="store",
     default=500,
     help="Number of features to be detected per image. "
-    "Only used for the detectors \'orb\' and \'sift'\. "
+    "Only used for the detectors 'orb' and 'sift'. "
     "The default is 500.",
     type=int,
 )

stitching/stitcher.py

@@ -48,7 +48,7 @@ def __init__(self, **kwargs):
         self.initialize_stitcher(**kwargs)
 
     def initialize_stitcher(self, **kwargs):
-        self.settings = Stitcher.DEFAULT_SETTINGS.copy()
+        self.settings = self.DEFAULT_SETTINGS.copy()
         self.validate_kwargs(kwargs)
         self.settings.update(kwargs)
 
@@ -240,10 +240,9 @@ def create_final_panorama(self):
             panorama, _ = self.blender.blend()
             return panorama
 
-    @staticmethod
-    def validate_kwargs(kwargs):
+    def validate_kwargs(self, kwargs):
         for arg in kwargs:
-            if arg not in Stitcher.DEFAULT_SETTINGS:
+            if arg not in self.DEFAULT_SETTINGS:
                 raise StitchingError("Invalid Argument: " + arg)
 
 
@@ -257,8 +256,5 @@ class AffineStitcher(Stitcher):
         "compensator": "no",
     }
 
-    def initialize_stitcher(self, **kwargs):
-        affine_defaults = AffineStitcher.AFFINE_DEFAULTS.copy()
-        self.validate_kwargs(kwargs)
-        affine_defaults.update(kwargs)
-        super().initialize_stitcher(**affine_defaults)
+    DEFAULT_SETTINGS = Stitcher.DEFAULT_SETTINGS.copy()
+    DEFAULT_SETTINGS.update(AFFINE_DEFAULTS)