fix(#133): hook should always be applied in all the frames

williamfzc · Aug 6, 2020 · 3184a0e · 3184a0e
1 parent 2d28a63
commit 3184a0e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stagesepx/classifier/base.py b/stagesepx/classifier/base.py
@@ -418,6 +418,8 @@ def classify(
         # for boost
         prev_result: typing.Optional[str] = None
         while frame is not None:
+            # hook
+            frame = self._apply_hook(frame, *args, **kwargs)
             # ignore some ranges
             if limit_range and not any(
                 [each.contain(frame.frame_id) for each in limit_range]
@@ -435,8 +437,6 @@ def classify(
                     result = prev_result
                 # else, do the real job
                 else:
-                    # hook
-                    frame = self._apply_hook(frame, *args, **kwargs)
                     prev_result = result = self._classify_frame(frame, *args, **kwargs)
                 logger.debug(
                     f"frame {frame.frame_id} ({frame.timestamp}) belongs to {result}"