fix: rename EmptyFrameDetectHook to InterestPointHook

with ORB detector
williamfzc · Sep 29, 2019 · be24630 · be24630
1 parent ba8ef2a
commit be24630
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/stagesepx/hook.py b/stagesepx/hook.py
@@ -244,7 +244,8 @@ def do(
         return
 
 
-class EmptyFrameDetectHook(BaseHook):
+class InterestPointHook(BaseHook):
+    """ use ORB detector to get the number of interest points """
     def __init__(self, *_, **__):
         super().__init__(*_, **__)
         self._orb = cv2.ORB_create()

diff --git a/test/test_hook.py b/test/test_hook.py
@@ -7,7 +7,7 @@
     CropHook,
     FrameSaveHook,
     RefineHook,
-    EmptyFrameDetectHook,
+    InterestPointHook,
     TemplateCompareHook,
 )
 
@@ -29,7 +29,7 @@ def test_hook():
     hook3 = FrameSaveHook(frame_home)
     hook4 = CropHook(size=(0.5, 0.5), offset=(0.0, 0.5), overwrite=True)
     hook5 = RefineHook()
-    hook6 = EmptyFrameDetectHook()
+    hook6 = InterestPointHook()
     hook7 = TemplateCompareHook({"amazon": IMAGE_PATH})
 
     # --- cutter ---