fix(preprocessor): fix type of index in shotdetect

gnes-ai · Sep 19, 2019 · 092379e · 092379e
1 parent 6833a27
commit 092379e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gnes/preprocessor/video/shotdetect.py b/gnes/preprocessor/video/shotdetect.py
@@ -99,7 +99,7 @@ def apply(self, doc: 'gnes_pb2.Document') -> None:
                     shot_len = len(frames)
                     c.weight = shot_len / num_frames
                     if self.sframes > 0 and shot_len > self.sframes:
-                        start_id = (shot_len - self.sframes) / 2
+                        start_id = int((shot_len - self.sframes) / 2)
                         end_id = start_id + self.sframes
                         frames = frames[start_id:end_id]
                     chunk_data = np.array(frames)