Fix index out bounds

futurice · Sep 22, 2019 · 238b079 · 238b079
1 parent 74eadbf
commit 238b079
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/drowsiness_detect.py b/drowsiness_detect.py
@@ -84,8 +84,7 @@ def get_next_image(image_name):
 def get_final_drowsiness_level(drowsiness_counts):
     l = lambda key: drowsiness_counts[key] >= EYE_ASPECT_RATIO_REQUIRED_FRAMES
     filtered = list(filter(l, drowsiness_counts.keys()))
-    print(filtered)
-    return drowsiness_to_number(filtered[0])
+    return drowsiness_to_number(filtered[0]) if len(filtered) > 0 else 0
 
 def drowsiness_to_number(drowsiness):
     return {