add GraphCutSeamFinder (#43)

* add GraphCutSeamFinder * bump version
OpenStitching · Jan 3, 2023 · 73bd810 · 73bd810
1 parent 81cb7c0
commit 73bd810
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -49,7 +49,7 @@ panorama = stitcher.stitch(["img1.jpg", "img2.jpg", "img3.jpg"])
 Specify your custom settings as
 
 ```python
-settings = {"detector": "sift", "confidence_threshold": 0.2}    
+settings = {"detector": "sift", "confidence_threshold": 0.2}
 stitcher = Stitcher(**settings)
 ```
 

diff --git a/stitching/__init__.py b/stitching/__init__.py
@@ -1,3 +1,3 @@
 from .stitcher import AffineStitcher, Stitcher  # noqa: F401
 
-__version__ = "0.3.0"
+__version__ = "0.4.0"
diff --git a/stitching/seam_finder.py b/stitching/seam_finder.py
@@ -12,12 +12,16 @@ class SeamFinder:
     SEAM_FINDER_CHOICES = OrderedDict()
     SEAM_FINDER_CHOICES["dp_color"] = cv.detail_DpSeamFinder("COLOR")
     SEAM_FINDER_CHOICES["dp_colorgrad"] = cv.detail_DpSeamFinder("COLOR_GRAD")
+    SEAM_FINDER_CHOICES["gc_color"] = cv.detail_GraphCutSeamFinder("COST_COLOR")
+    SEAM_FINDER_CHOICES["gc_colorgrad"] = cv.detail_GraphCutSeamFinder(
+        "COST_COLOR_GRAD"
+    )
     SEAM_FINDER_CHOICES["voronoi"] = cv.detail.SeamFinder_createDefault(
         cv.detail.SeamFinder_VORONOI_SEAM
-    )  # noqa
+    )
     SEAM_FINDER_CHOICES["no"] = cv.detail.SeamFinder_createDefault(
         cv.detail.SeamFinder_NO
-    )  # noqa
+    )
 
     DEFAULT_SEAM_FINDER = list(SEAM_FINDER_CHOICES.keys())[0]