fix(shot-detector): fix case of only one shot in video

gnes-ai · Sep 5, 2019 · d70e877 · d70e877
1 parent 93a43f5
commit d70e877
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gnes/preprocessor/helper.py b/gnes/preprocessor/helper.py
@@ -358,6 +358,9 @@ def motion_algo(distances: List[float], **kwargs) -> List[int]:
     shots = []
     num_frames = len(distances) + 1
     p = peakutils.indexes(np.array(distances).astype('float32'), thres=arg_dict['threshold'], min_dist=arg_dict['min_dist'])
+    if len(p) == 0:
+        return [0, num_frames]
+
     shots.append(0)
     shots.append(p[0] + 2)
     for i in range(1, len(p)):