diff --git a/stitching/feature_detector.py b/stitching/feature_detector.py index 9cf7112..1435655 100644 --- a/stitching/feature_detector.py +++ b/stitching/feature_detector.py @@ -13,8 +13,6 @@ class FeatureDetector: DETECTOR_CHOICES["orb"] = cv.ORB.create DETECTOR_CHOICES["sift"] = cv.SIFT_create - DETECTOR_CHOICES["brisk"] = cv.BRISK_create - DETECTOR_CHOICES["akaze"] = cv.AKAZE_create DEFAULT_DETECTOR = list(DETECTOR_CHOICES.keys())[0] diff --git a/tests/stitching_detailed.py b/tests/stitching_detailed.py index 6b1d0b8..8456879 100644 --- a/tests/stitching_detailed.py +++ b/tests/stitching_detailed.py @@ -39,11 +39,13 @@ except AttributeError: pass try: - FEATURES_FIND_CHOICES["brisk"] = cv.BRISK_create + cv.xfeatures2d_BRISK.create() # check if the function can be called + FEATURES_FIND_CHOICES["brisk"] = cv.xfeatures2d_BRISK.create except AttributeError: print("BRISK not available") try: - FEATURES_FIND_CHOICES["akaze"] = cv.AKAZE_create + cv.xfeatures2d_AKAZE.create() # check if the function can be called + FEATURES_FIND_CHOICES["akaze"] = cv.xfeatures2d_AKAZE.create except AttributeError: print("AKAZE not available")