From bfbcc9ca6d7dde12aebb6d534cb4d5ef1f08fd40 Mon Sep 17 00:00:00 2001 From: Sebastian Fitt Date: Fri, 24 May 2024 21:54:58 +0200 Subject: [PATCH] fix: Division by zero --- EyeTrackApp/ellipse_based_pupil_dilation.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/EyeTrackApp/ellipse_based_pupil_dilation.py b/EyeTrackApp/ellipse_based_pupil_dilation.py index 46a4d0b..d9fd53a 100644 --- a/EyeTrackApp/ellipse_based_pupil_dilation.py +++ b/EyeTrackApp/ellipse_based_pupil_dilation.py @@ -350,12 +350,10 @@ class EllipseBasedPupilDilation: minp = float(self.maxval) try: - if maxp != 0 and not (np.isnan(pupil_area) or np.isnan(maxp)): - eyedilation = (pupil_area - maxp) / ( - minp - maxp - ) - else: + if not np.isfinite(pupil_area) or not np.isfinite(maxp) or not np.isfinite(minp) or (minp - maxp) == 0: eyedilation = 0.5 + else: + eyedilation = (pupil_area - maxp) / (minp - maxp) except: eyedilation = 0.5 eyedilation = 1 - eyedilation