From 238b079a3152be79ac0cb7599fcf10f8da0750b0 Mon Sep 17 00:00:00 2001 From: Joonas Sarapalo Date: Sun, 22 Sep 2019 09:13:22 +0300 Subject: [PATCH] Fix index out bounds --- drowsiness_detect.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/drowsiness_detect.py b/drowsiness_detect.py index ad43a24..d17e521 100644 --- 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 {