add config dictionary to get_depth.py

2025-12-30 10:25:16 +08:00
parent b9573c5781
commit 93e2984e7f
6 changed files with 358 additions and 334 deletions
@@ -6,15 +6,15 @@ import cv2

 class Stitcher:

-    def __init__(self,isstitch=False, n=[3,3], index=np.array([1,2,3,6,7,8,11,12,13])):
+    def __init__(self,isstitch=False, n=[3,3], index=np.array([1,2,3,6,7,8,11,12,13]), config_stitcher=None):

        # Settings
        self.isstitch = isstitch
-
+        self.config_stitcher = config_stitcher
        # Find Boxes Related
-        self.min_area_frac = 0.01
-        self.max_area_frac = 0.05
-        self.max_aspect_ratio = 100
+        self.min_area_frac = config_stitcher["min_area_frac"]
+        self.max_area_frac = config_stitcher["max_area_frac"]
+        self.max_aspect_ratio = config_stitcher["max_aspect_ratio"]

        # stitch related
        self.stitch_valid = False