diff --git a/EyeTrackApp/ellipse_based_pupil_dilation.py b/EyeTrackApp/ellipse_based_pupil_dilation.py index 440c04a..3bb8839 100644 --- a/EyeTrackApp/ellipse_based_pupil_dilation.py +++ b/EyeTrackApp/ellipse_based_pupil_dilation.py @@ -359,9 +359,10 @@ class EllipseBasedPupilDilation: minp = float(self.maxval) try: - eyedilation = (pupil_area - maxp) / ( - minp - maxp - ) # for whatever reason when input and maxp are too close it outputs high + if maxp != 0 and not (np.isnan(pupil_area) or np.isnan(maxp)): + eyedilation = (pupil_area - maxp) / ( + minp - maxp + ) # for whatever reason when input and maxp are too close it outputs high except: eyedilation = 0.5 eyedilation = 1 - eyedilation