mirror of
https://github.com/EyeTrackVR/EyeTrackVR.git
synced 2025-11-04 14:39:42 +08:00
black format ep
This commit is contained in:
parent
635173f42a
commit
c784918531
@ -1,4 +1,4 @@
|
|||||||
'''
|
"""
|
||||||
------------------------------------------------------------------------------------------------------
|
------------------------------------------------------------------------------------------------------
|
||||||
|
|
||||||
,@@@@@@
|
,@@@@@@
|
||||||
@ -28,7 +28,7 @@ Additional Contributors: [Assassin], Summer404NotFound, lorow, ZanzyTHEbar
|
|||||||
|
|
||||||
Copyright (c) 2023 EyeTrackVR <3
|
Copyright (c) 2023 EyeTrackVR <3
|
||||||
------------------------------------------------------------------------------------------------------
|
------------------------------------------------------------------------------------------------------
|
||||||
'''
|
"""
|
||||||
|
|
||||||
from operator import truth
|
from operator import truth
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
@ -36,7 +36,6 @@ import sys
|
|||||||
import asyncio
|
import asyncio
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
sys.path.append(".")
|
sys.path.append(".")
|
||||||
from config import EyeTrackCameraConfig
|
from config import EyeTrackCameraConfig
|
||||||
from config import EyeTrackSettingsConfig
|
from config import EyeTrackSettingsConfig
|
||||||
@ -61,6 +60,7 @@ from utils.img_utils import circle_crop
|
|||||||
from eye import EyeInfo, EyeInfoOrigin
|
from eye import EyeInfo, EyeInfoOrigin
|
||||||
from intensity_based_openness import *
|
from intensity_based_openness import *
|
||||||
|
|
||||||
|
|
||||||
def run_once(f):
|
def run_once(f):
|
||||||
def wrapper(*args, **kwargs):
|
def wrapper(*args, **kwargs):
|
||||||
if not wrapper.has_run:
|
if not wrapper.has_run:
|
||||||
@ -74,8 +74,7 @@ def run_once(f):
|
|||||||
async def delayed_setting_change(setting, value):
|
async def delayed_setting_change(setting, value):
|
||||||
await asyncio.sleep(5)
|
await asyncio.sleep(5)
|
||||||
setting = value
|
setting = value
|
||||||
PlaySound('Audio/completed.wav', SND_FILENAME | SND_ASYNC)
|
PlaySound("Audio/completed.wav", SND_FILENAME | SND_ASYNC)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class EyeProcessor:
|
class EyeProcessor:
|
||||||
@ -148,7 +147,7 @@ class EyeProcessor:
|
|||||||
self.rawx = 0.0
|
self.rawx = 0.0
|
||||||
self.rawy = 0.0
|
self.rawy = 0.0
|
||||||
self.eyeopen = 0.7
|
self.eyeopen = 0.7
|
||||||
#blink
|
# blink
|
||||||
self.max_ints = []
|
self.max_ints = []
|
||||||
self.max_int = 0
|
self.max_int = 0
|
||||||
self.min_int = 4000000000000
|
self.min_int = 4000000000000
|
||||||
@ -162,19 +161,16 @@ class EyeProcessor:
|
|||||||
self.bd_blink = False
|
self.bd_blink = False
|
||||||
self.current_algo = EyeInfoOrigin.HSRAC
|
self.current_algo = EyeInfoOrigin.HSRAC
|
||||||
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
min_cutoff = float(self.settings.gui_min_cutoff) # 0.0004
|
min_cutoff = float(self.settings.gui_min_cutoff) # 0.0004
|
||||||
beta = float(self.settings.gui_speed_coefficient) # 0.9
|
beta = float(self.settings.gui_speed_coefficient) # 0.9
|
||||||
except:
|
except:
|
||||||
print('\033[93m[WARN] OneEuroFilter values must be a legal number.\033[0m')
|
print("\033[93m[WARN] OneEuroFilter values must be a legal number.\033[0m")
|
||||||
min_cutoff = 0.0004
|
min_cutoff = 0.0004
|
||||||
beta = 0.9
|
beta = 0.9
|
||||||
noisy_point = np.array([1, 1])
|
noisy_point = np.array([1, 1])
|
||||||
self.one_euro_filter = OneEuroFilter(
|
self.one_euro_filter = OneEuroFilter(
|
||||||
noisy_point,
|
noisy_point, min_cutoff=min_cutoff, beta=beta
|
||||||
min_cutoff=min_cutoff,
|
|
||||||
beta=beta
|
|
||||||
)
|
)
|
||||||
|
|
||||||
def output_images_and_update(self, threshold_image, output_information: EyeInfo):
|
def output_images_and_update(self, threshold_image, output_information: EyeInfo):
|
||||||
@ -189,20 +185,23 @@ class EyeProcessor:
|
|||||||
self.image_queue_outgoing.put((image_stack, output_information))
|
self.image_queue_outgoing.put((image_stack, output_information))
|
||||||
self.previous_image = self.current_image
|
self.previous_image = self.current_image
|
||||||
self.previous_rotation = self.config.rotation_angle
|
self.previous_rotation = self.config.rotation_angle
|
||||||
except: # If this fails it likely means that the images are not the same size for some reason.
|
except: # If this fails it likely means that the images are not the same size for some reason.
|
||||||
print('\033[91m[ERROR] Size of frames to display are of unequal sizes.\033[0m')
|
print(
|
||||||
|
"\033[91m[ERROR] Size of frames to display are of unequal sizes.\033[0m"
|
||||||
|
)
|
||||||
|
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def capture_crop_rotate_image(self):
|
def capture_crop_rotate_image(self):
|
||||||
# Get our current frame
|
# Get our current frame
|
||||||
|
|
||||||
try:
|
try:
|
||||||
# Get frame from capture source, crop to ROI
|
# Get frame from capture source, crop to ROI
|
||||||
self.current_image = self.current_image[
|
self.current_image = self.current_image[
|
||||||
int(self.config.roi_window_y): int(
|
int(self.config.roi_window_y) : int(
|
||||||
self.config.roi_window_y + self.config.roi_window_h
|
self.config.roi_window_y + self.config.roi_window_h
|
||||||
),
|
),
|
||||||
int(self.config.roi_window_x): int(
|
int(self.config.roi_window_x) : int(
|
||||||
self.config.roi_window_x + self.config.roi_window_w
|
self.config.roi_window_x + self.config.roi_window_w
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
@ -232,7 +231,7 @@ class EyeProcessor:
|
|||||||
rotation_matrix,
|
rotation_matrix,
|
||||||
(cols, rows),
|
(cols, rows),
|
||||||
borderMode=cv2.BORDER_CONSTANT,
|
borderMode=cv2.BORDER_CONSTANT,
|
||||||
borderValue=(ar + 10, ag + 10, ab + 10),#(255, 255, 255),
|
borderValue=(ar + 10, ag + 10, ab + 10), # (255, 255, 255),
|
||||||
)
|
)
|
||||||
self.current_image_white = cv2.warpAffine(
|
self.current_image_white = cv2.warpAffine(
|
||||||
self.current_image,
|
self.current_image,
|
||||||
@ -245,20 +244,33 @@ class EyeProcessor:
|
|||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
def UPDATE(self):
|
def UPDATE(self):
|
||||||
if self.settings.gui_BLINK:
|
if self.settings.gui_BLINK:
|
||||||
self.eyeopen = BLINK(self)
|
self.eyeopen = BLINK(self)
|
||||||
|
|
||||||
if self.settings.gui_IBO:
|
if self.settings.gui_IBO:
|
||||||
self.eyeopen = self.ibo.intense(self.rawx, self.rawy, self.current_image_white,self.settings.ibo_filter_samples,self.settings.ibo_average_output_samples)
|
self.eyeopen = self.ibo.intense(
|
||||||
if self.eyeopen < self.settings.ibo_fully_close_eye_threshold: #threshold so the eye fully closes
|
self.rawx,
|
||||||
|
self.rawy,
|
||||||
|
self.current_image_white,
|
||||||
|
self.settings.ibo_filter_samples,
|
||||||
|
self.settings.ibo_average_output_samples,
|
||||||
|
)
|
||||||
|
if (
|
||||||
|
self.eyeopen < self.settings.ibo_fully_close_eye_threshold
|
||||||
|
): # threshold so the eye fully closes
|
||||||
self.eyeopen = 0.0
|
self.eyeopen = 0.0
|
||||||
if self.bd_blink == True:
|
if self.bd_blink == True:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
if self.settings.gui_IBO and self.settings.gui_BLINK:
|
if self.settings.gui_IBO and self.settings.gui_BLINK:
|
||||||
ibo = self.ibo.intense(self.rawx, self.rawy, self.current_image_white,self.settings.ibo_filter_samples,self.settings.ibo_average_output_samples)
|
ibo = self.ibo.intense(
|
||||||
|
self.rawx,
|
||||||
|
self.rawy,
|
||||||
|
self.current_image_white,
|
||||||
|
self.settings.ibo_filter_samples,
|
||||||
|
self.settings.ibo_average_output_samples,
|
||||||
|
)
|
||||||
|
|
||||||
blink = BLINK(self)
|
blink = BLINK(self)
|
||||||
if blink == 0.0:
|
if blink == 0.0:
|
||||||
@ -266,8 +278,10 @@ class EyeProcessor:
|
|||||||
else:
|
else:
|
||||||
self.eyeopen = ibo
|
self.eyeopen = ibo
|
||||||
|
|
||||||
self.output_images_and_update(self.thresh, EyeInfo(self.current_algo, self.out_x, self.out_y, 0, self.eyeopen))
|
self.output_images_and_update(
|
||||||
|
self.thresh,
|
||||||
|
EyeInfo(self.current_algo, self.out_x, self.out_y, 0, self.eyeopen),
|
||||||
|
)
|
||||||
|
|
||||||
def BLINKM(self):
|
def BLINKM(self):
|
||||||
self.eyeopen = BLINK(self)
|
self.eyeopen = BLINK(self)
|
||||||
@ -275,40 +289,51 @@ class EyeProcessor:
|
|||||||
def DADDYM(self):
|
def DADDYM(self):
|
||||||
# todo: We should have a proper variable for drawing.
|
# todo: We should have a proper variable for drawing.
|
||||||
self.thresh = self.current_image_gray.copy()
|
self.thresh = self.current_image_gray.copy()
|
||||||
self.rawx, self.rawy, self.eyeopen, self.radius = self.er_daddy.run(self.current_image_gray)
|
self.rawx, self.rawy, self.eyeopen, self.radius = self.er_daddy.run(
|
||||||
|
self.current_image_gray
|
||||||
|
)
|
||||||
# Daddy also uses a one euro filter, so I'll have to use it twice, but I'm not going to think too much about it.
|
# Daddy also uses a one euro filter, so I'll have to use it twice, but I'm not going to think too much about it.
|
||||||
self.out_x, self.out_y = cal.cal_osc(self, self.rawx, self.rawy)
|
self.out_x, self.out_y = cal.cal_osc(self, self.rawx, self.rawy)
|
||||||
self.current_algorithm = EyeInfoOrigin.DADDY
|
self.current_algorithm = EyeInfoOrigin.DADDY
|
||||||
|
|
||||||
def HSRACM(self):
|
def HSRACM(self):
|
||||||
if self.eye_id in [EyeId.LEFT] and self.settings.gui_circular_crop_left:
|
if self.eye_id in [EyeId.LEFT] and self.settings.gui_circular_crop_left:
|
||||||
self.current_image_gray, self.cct = circle_crop(self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct)
|
self.current_image_gray, self.cct = circle_crop(
|
||||||
|
self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
if self.eye_id in [EyeId.RIGHT] and self.settings.gui_circular_crop_right:
|
if self.eye_id in [EyeId.RIGHT] and self.settings.gui_circular_crop_right:
|
||||||
self.current_image_gray, self.cct = circle_crop(self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct)
|
self.current_image_gray, self.cct = circle_crop(
|
||||||
|
self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
self.hasrac_en = True
|
self.hasrac_en = True
|
||||||
# todo: add process to initialise er_hsrac when resolution changes
|
# todo: add process to initialise er_hsrac when resolution changes
|
||||||
self.rawx, self.rawy, self.thresh, self.radius = self.er_hsf.run(self.current_image_gray)
|
self.rawx, self.rawy, self.thresh, self.radius = self.er_hsf.run(
|
||||||
|
self.current_image_gray
|
||||||
|
)
|
||||||
self.rawx, self.rawy, self.thresh = RANSAC3D(self, True)
|
self.rawx, self.rawy, self.thresh = RANSAC3D(self, True)
|
||||||
#print(self.radius)
|
# print(self.radius)
|
||||||
#if self.prev_x is None:
|
# if self.prev_x is None:
|
||||||
# self.prev_x = self.rawx
|
# self.prev_x = self.rawx
|
||||||
# self.prev_y = self.rawy
|
# self.prev_y = self.rawy
|
||||||
self.out_x, self.out_y = cal.cal_osc(self, self.rawx, self.rawy)
|
self.out_x, self.out_y = cal.cal_osc(self, self.rawx, self.rawy)
|
||||||
self.current_algorithm = EyeInfoOrigin.HSRAC
|
self.current_algorithm = EyeInfoOrigin.HSRAC
|
||||||
|
|
||||||
|
|
||||||
def HSFM(self):
|
def HSFM(self):
|
||||||
if self.eye_id in [EyeId.LEFT] and self.settings.gui_circular_crop_left:
|
if self.eye_id in [EyeId.LEFT] and self.settings.gui_circular_crop_left:
|
||||||
self.current_image_gray, self.cct = circle_crop(self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct)
|
self.current_image_gray, self.cct = circle_crop(
|
||||||
|
self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
if self.eye_id in [EyeId.RIGHT] and self.settings.gui_circular_crop_right:
|
if self.eye_id in [EyeId.RIGHT] and self.settings.gui_circular_crop_right:
|
||||||
self.current_image_gray, self.cct = circle_crop(self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct)
|
self.current_image_gray, self.cct = circle_crop(
|
||||||
|
self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
# todo: add process to initialise er_hsf when resolution changes
|
# todo: add process to initialise er_hsf when resolution changes
|
||||||
@ -318,26 +343,36 @@ class EyeProcessor:
|
|||||||
|
|
||||||
def RANSAC3DM(self):
|
def RANSAC3DM(self):
|
||||||
if self.eye_id in [EyeId.LEFT] and self.settings.gui_circular_crop_left:
|
if self.eye_id in [EyeId.LEFT] and self.settings.gui_circular_crop_left:
|
||||||
self.current_image_gray, self.cct = circle_crop(self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct)
|
self.current_image_gray, self.cct = circle_crop(
|
||||||
|
self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
if self.eye_id in [EyeId.RIGHT] and self.settings.gui_circular_crop_right:
|
if self.eye_id in [EyeId.RIGHT] and self.settings.gui_circular_crop_right:
|
||||||
self.current_image_gray, self.cct = circle_crop(self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct)
|
self.current_image_gray, self.cct = circle_crop(
|
||||||
|
self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
self.hasrac_en = False
|
self.hasrac_en = False
|
||||||
current_image_gray_copy = self.current_image_gray.copy() # Duplicate before overwriting in RANSAC3D.
|
current_image_gray_copy = (
|
||||||
|
self.current_image_gray.copy()
|
||||||
|
) # Duplicate before overwriting in RANSAC3D.
|
||||||
self.rawx, self.rawy, self.thresh = RANSAC3D(self, False)
|
self.rawx, self.rawy, self.thresh = RANSAC3D(self, False)
|
||||||
self.out_x, self.out_y = cal.cal_osc(self, self.rawx, self.rawy)
|
self.out_x, self.out_y = cal.cal_osc(self, self.rawx, self.rawy)
|
||||||
self.current_algorithm = EyeInfoOrigin.RANSAC
|
self.current_algorithm = EyeInfoOrigin.RANSAC
|
||||||
|
|
||||||
def BLOBM(self):
|
def BLOBM(self):
|
||||||
if self.eye_id in [EyeId.LEFT] and self.settings.gui_circular_crop_left:
|
if self.eye_id in [EyeId.LEFT] and self.settings.gui_circular_crop_left:
|
||||||
self.current_image_gray, self.cct = circle_crop(self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct)
|
self.current_image_gray, self.cct = circle_crop(
|
||||||
|
self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
if self.eye_id in [EyeId.RIGHT] and self.settings.gui_circular_crop_right:
|
if self.eye_id in [EyeId.RIGHT] and self.settings.gui_circular_crop_right:
|
||||||
self.current_image_gray, self.cct = circle_crop(self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct)
|
self.current_image_gray, self.cct = circle_crop(
|
||||||
|
self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
self.rawx, self.rawy, self.thresh = BLOB(self)
|
self.rawx, self.rawy, self.thresh = BLOB(self)
|
||||||
@ -345,8 +380,6 @@ class EyeProcessor:
|
|||||||
self.out_x, self.out_y = cal.cal_osc(self, self.rawx, self.rawy)
|
self.out_x, self.out_y = cal.cal_osc(self, self.rawx, self.rawy)
|
||||||
self.current_algorithm = EyeInfoOrigin.BLOB
|
self.current_algorithm = EyeInfoOrigin.BLOB
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def ALGOSELECT(self):
|
def ALGOSELECT(self):
|
||||||
# self.DADDYM()
|
# self.DADDYM()
|
||||||
if self.failed == 0 and self.firstalgo != None:
|
if self.failed == 0 and self.firstalgo != None:
|
||||||
@ -354,7 +387,9 @@ class EyeProcessor:
|
|||||||
else:
|
else:
|
||||||
self.failed = self.failed + 1
|
self.failed = self.failed + 1
|
||||||
|
|
||||||
if self.failed == 1 and self.secondalgo != None: #send the tracking algos previous fail number, in algo if we pass set to 0, if fail, + 1
|
if (
|
||||||
|
self.failed == 1 and self.secondalgo != None
|
||||||
|
): # send the tracking algos previous fail number, in algo if we pass set to 0, if fail, + 1
|
||||||
self.secondalgo()
|
self.secondalgo()
|
||||||
else:
|
else:
|
||||||
self.failed = self.failed + 1
|
self.failed = self.failed + 1
|
||||||
@ -373,17 +408,12 @@ class EyeProcessor:
|
|||||||
self.fithalgo()
|
self.fithalgo()
|
||||||
|
|
||||||
else:
|
else:
|
||||||
self.failed = 0 # we have reached last possible algo and it is disabled, move to first algo
|
self.failed = 0 # we have reached last possible algo and it is disabled, move to first algo
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
# Run the following somewhere
|
# Run the following somewhere
|
||||||
# self.daddy = External_Run_DADDY()
|
# self.daddy = External_Run_DADDY()
|
||||||
|
|
||||||
|
|
||||||
self.firstalgo = None
|
self.firstalgo = None
|
||||||
self.secondalgo = None
|
self.secondalgo = None
|
||||||
self.thirdalgo = None
|
self.thirdalgo = None
|
||||||
@ -391,19 +421,24 @@ class EyeProcessor:
|
|||||||
self.fithalgo = None
|
self.fithalgo = None
|
||||||
algolist = [None, None, None, None, None, None]
|
algolist = [None, None, None, None, None, None]
|
||||||
|
|
||||||
#clear HSF values when page is opened to correctly reflect setting changes
|
# clear HSF values when page is opened to correctly reflect setting changes
|
||||||
self.er_hsf = None
|
self.er_hsf = None
|
||||||
|
|
||||||
#set algo priorities
|
# set algo priorities
|
||||||
if self.settings.gui_HSF:
|
if self.settings.gui_HSF:
|
||||||
if self.er_hsf is None:
|
if self.er_hsf is None:
|
||||||
if self.eye_id in [EyeId.LEFT]:
|
if self.eye_id in [EyeId.LEFT]:
|
||||||
self.er_hsf = External_Run_HSF(self.settings.gui_skip_autoradius, self.settings.gui_HSF_radius_left)
|
self.er_hsf = External_Run_HSF(
|
||||||
|
self.settings.gui_skip_autoradius,
|
||||||
|
self.settings.gui_HSF_radius_left,
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
if self.eye_id in [EyeId.RIGHT]:
|
if self.eye_id in [EyeId.RIGHT]:
|
||||||
self.er_hsf = External_Run_HSF(self.settings.gui_skip_autoradius,
|
self.er_hsf = External_Run_HSF(
|
||||||
self.settings.gui_HSF_radius_right)
|
self.settings.gui_skip_autoradius,
|
||||||
|
self.settings.gui_HSF_radius_right,
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
algolist[self.settings.gui_HSFP] = self.HSFM
|
algolist[self.settings.gui_HSFP] = self.HSFM
|
||||||
@ -416,12 +451,17 @@ class EyeProcessor:
|
|||||||
|
|
||||||
if self.eye_id in [EyeId.LEFT]:
|
if self.eye_id in [EyeId.LEFT]:
|
||||||
|
|
||||||
self.er_hsf = External_Run_HSF(self.settings.gui_skip_autoradius, self.settings.gui_HSF_radius_left)
|
self.er_hsf = External_Run_HSF(
|
||||||
|
self.settings.gui_skip_autoradius,
|
||||||
|
self.settings.gui_HSF_radius_left,
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
if self.eye_id in [EyeId.RIGHT]:
|
if self.eye_id in [EyeId.RIGHT]:
|
||||||
self.er_hsf = External_Run_HSF(self.settings.gui_skip_autoradius,
|
self.er_hsf = External_Run_HSF(
|
||||||
self.settings.gui_HSF_radius_right)
|
self.settings.gui_skip_autoradius,
|
||||||
|
self.settings.gui_HSF_radius_right,
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
pass
|
pass
|
||||||
algolist[self.settings.gui_HSRACP] = self.HSRACM
|
algolist[self.settings.gui_HSRACP] = self.HSRACM
|
||||||
@ -443,19 +483,23 @@ class EyeProcessor:
|
|||||||
if self.settings.gui_BLOB:
|
if self.settings.gui_BLOB:
|
||||||
algolist[self.settings.gui_BLOBP] = self.BLOBM
|
algolist[self.settings.gui_BLOBP] = self.BLOBM
|
||||||
|
|
||||||
_, self.firstalgo, self.secondalgo, self.thirdalgo, self.fourthalgo, self.fithalgo = algolist
|
(
|
||||||
|
_,
|
||||||
|
self.firstalgo,
|
||||||
|
self.secondalgo,
|
||||||
|
self.thirdalgo,
|
||||||
|
self.fourthalgo,
|
||||||
|
self.fithalgo,
|
||||||
|
) = algolist
|
||||||
|
|
||||||
f = True
|
f = True
|
||||||
while True:
|
while True:
|
||||||
# f = True
|
# f = True
|
||||||
# Check to make sure we haven't been requested to close
|
# Check to make sure we haven't been requested to close
|
||||||
if self.cancellation_event.is_set():
|
if self.cancellation_event.is_set():
|
||||||
print("\033[94m[INFO] Exiting Tracking thread\033[0m")
|
print("\033[94m[INFO] Exiting Tracking thread\033[0m")
|
||||||
return
|
return
|
||||||
|
|
||||||
|
|
||||||
if self.config.roi_window_w <= 0 or self.config.roi_window_h <= 0:
|
if self.config.roi_window_w <= 0 or self.config.roi_window_h <= 0:
|
||||||
# At this point, we're waiting for the user to set up the ROI window in the GUI.
|
# At this point, we're waiting for the user to set up the ROI window in the GUI.
|
||||||
# Sleep a bit while we wait.
|
# Sleep a bit while we wait.
|
||||||
@ -463,11 +507,12 @@ class EyeProcessor:
|
|||||||
return
|
return
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
|
||||||
# If our ROI configuration has changed, reset our model and detector
|
# If our ROI configuration has changed, reset our model and detector
|
||||||
if (self.camera_model is None
|
if (
|
||||||
|
self.camera_model is None
|
||||||
or self.detector_3d is None
|
or self.detector_3d is None
|
||||||
or self.camera_model.resolution != (
|
or self.camera_model.resolution
|
||||||
|
!= (
|
||||||
self.config.roi_window_w,
|
self.config.roi_window_w,
|
||||||
self.config.roi_window_h,
|
self.config.roi_window_h,
|
||||||
)
|
)
|
||||||
@ -496,12 +541,12 @@ class EyeProcessor:
|
|||||||
if not self.capture_crop_rotate_image():
|
if not self.capture_crop_rotate_image():
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
|
||||||
self.current_image_gray = cv2.cvtColor(
|
self.current_image_gray = cv2.cvtColor(
|
||||||
self.current_image, cv2.COLOR_BGR2GRAY
|
self.current_image, cv2.COLOR_BGR2GRAY
|
||||||
)
|
)
|
||||||
self.current_image_gray_clean = self.current_image_gray.copy() #copy this frame to have a clean image for blink algo
|
self.current_image_gray_clean = (
|
||||||
|
self.current_image_gray.copy()
|
||||||
|
) # copy this frame to have a clean image for blink algo
|
||||||
|
|
||||||
self.ALGOSELECT() #run our algos in priority order set in settings
|
self.ALGOSELECT() # run our algos in priority order set in settings
|
||||||
self.UPDATE()
|
self.UPDATE()
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user