fix(#153): wrong calc in CropHook

williamfzc · Feb 1, 2021 · 72d47df · 72d47df
1 parent 2b7235f
commit 72d47df
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/stagesepx/hook.py b/stagesepx/hook.py
@@ -170,10 +170,10 @@ def do(self, frame: VideoFrame, *_, **__) -> typing.Optional[VideoFrame]:
         height_range, width_range = self.convert_size_and_offset(*frame.data.shape)
         # ignore the rest of this frame
         # same as IgnoreHook
-        frame.data[: height_range[0]] = 0
-        frame.data[height_range[1] :] = 0
-        frame.data[: width_range[0]] = 0
-        frame.data[width_range[1] :] = 0
+        frame.data[: height_range[0], :] = 0
+        frame.data[height_range[1] :, :] = 0
+        frame.data[:, : width_range[0]] = 0
+        frame.data[:, width_range[1] :] = 0
         return frame