feat(preprocessor): add frame selector

gnes-ai · Sep 25, 2019 · 09199d8 · 09199d8
1 parent 80364c0
commit 09199d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/gnes/preprocessor/video/frame_select.py b/gnes/preprocessor/video/frame_select.py
@@ -35,7 +35,7 @@ def apply(self, doc: 'gnes_pb2.Document') -> None:
             for chunk in doc.chunks:
                 images = blob2array(chunk.blob)
                 if len(images) == 0:
-                    self.logger.info("this chunk has no frame!")
+                    self.logger.warning("this chunk has no frame!")
                 elif self.sframes == 1:
                     idx = [int(len(images) / 2)]
                     chunk.blob.CopyFrom(array2blob(images[idx]))

diff --git a/gnes/preprocessor/video/shotdetect.py b/gnes/preprocessor/video/shotdetect.py
@@ -106,7 +106,7 @@ def apply(self, doc: 'gnes_pb2.Document') -> None:
                         else:
                             idx = np.sort(np.random.choice(shot_len, self.sframes, replace=False))
                             frames = [frames[idx_] for idx_ in idx]
-                            
+
                     chunk_data = np.array(frames)
                     c.blob.CopyFrom(array2blob(chunk_data))
             else: