fix(preprocessor): fix bug when num_frames < 4 in shotdetect

gnes-ai · Sep 20, 2019 · 64acb4c · 64acb4c
1 parent af879ba
commit 64acb4c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gnes/preprocessor/helper.py b/gnes/preprocessor/helper.py
@@ -357,7 +357,7 @@ 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'])
+    p = peakutils.indexes(np.array(distances).astype('float32'), thres=arg_dict['threshold'], min_dist=arg_dict['min_dist']) if len(distances) else []
     if len(p) == 0:
         return [0, num_frames]