fix(#82): resize the video length after load_frames

williamfzc · Dec 17, 2019 · 1817cb7 · 1817cb7
1 parent 9ef6a3d
commit 1817cb7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/stagesepx/video.py b/stagesepx/video.py
@@ -134,6 +134,8 @@ def load_frames(self):
 
         # lock the order
         self.data = tuple(data)
+        # fix the length ( the last frame may be broken sometimes )
+        self.frame_count = len(data)
 
     def _read_from_file(self) -> typing.Generator[VideoFrame, None, None]:
         with toolbox.video_capture(self.path) as cap: