Add confidence threshold to camera adjuster

Author: chinery

stitching/camera_adjuster.py

@@ -19,12 +19,15 @@ class CameraAdjuster:
     DEFAULT_REFINEMENT_MASK = "xxxxx"
 
     def __init__(
-        self, adjuster=DEFAULT_CAMERA_ADJUSTER, refinement_mask=DEFAULT_REFINEMENT_MASK
+        self,
+        adjuster=DEFAULT_CAMERA_ADJUSTER,
+        refinement_mask=DEFAULT_REFINEMENT_MASK,
+        confidence_threshold=1.0,
     ):
 
         self.adjuster = CameraAdjuster.CAMERA_ADJUSTER_CHOICES[adjuster]()
         self.set_refinement_mask(refinement_mask)
-        self.adjuster.setConfThresh(1)
+        self.adjuster.setConfThresh(confidence_threshold)
 
     def set_refinement_mask(self, refinement_mask):
         mask_matrix = np.zeros((3, 3), np.uint8)

stitching/stitcher.py

@@ -71,7 +71,9 @@ def initialize_stitcher(self, **kwargs):
             args.confidence_threshold, args.matches_graph_dot_file
         )
         self.camera_estimator = CameraEstimator(args.estimator)
-        self.camera_adjuster = CameraAdjuster(args.adjuster, args.refinement_mask)
+        self.camera_adjuster = CameraAdjuster(
+            args.adjuster, args.refinement_mask, args.confidence_threshold
+        )
         self.wave_corrector = WaveCorrector(args.wave_correct_kind)
         self.warper = Warper(args.warper_type)
         self.cropper = Cropper(args.crop)