From dc6fbd34d1ff4349fc5d2510a12829bf2f76426e Mon Sep 17 00:00:00 2001 From: Prohurtz <48768484+RedHawk989@users.noreply.github.com> Date: Mon, 20 Nov 2023 12:44:34 -0600 Subject: [PATCH] fix: blur crash --- EyeTrackApp/ransac.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/EyeTrackApp/ransac.py b/EyeTrackApp/ransac.py index 3a1794e..a7b25ca 100644 --- a/EyeTrackApp/ransac.py +++ b/EyeTrackApp/ransac.py @@ -267,8 +267,12 @@ def RANSAC3D(self, hsrac_en): # Crop first to reduce the amount of data to process. # frame = frame[0:len(frame) - 5, :] # To reduce the processing data, blur. - - frame_gray = cv2.GaussianBlur(frame, (5, 5), 0) + if frame is None: + print("[WARN] Frame is empty") + self.failed = self.failed + 1 # we have failed, move onto next algo + return 0, 0, frame, blink, 0, 0 + else: + frame_gray = cv2.GaussianBlur(frame, (5, 5), 0) # this will need to be adjusted everytime hardware is changed (brightness of IR, Camera postion, etc)m min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(frame_gray)