Don't try to predict no stop for short tracks

Kalior · Aug 29, 2018 · 694fb8c · 694fb8c
1 parent 3fd5a4d
commit 694fb8c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/live_prediction.py b/live_prediction.py
@@ -143,7 +143,10 @@ def write_chunk_to_file(video_name, video, frames, chunk, label, out_dir, i):
     ChunkVisualiser().chunk_to_video_scene(video, chunk, out_file, frames, label)
 
 
-def predict_no_stop(track, confidence_threshold, stop_threshold=10):
+def predict_no_stop(track, confidence_threshold):
+    if len(track) < 50:
+        return False, ()
+
     classifier_prediction = classifier_predict_no_stop(track, confidence_threshold)
 
     #  Copy last 200 frames to chunk for visusalisation.