mirror of
https://github.com/EyeTrackVR/EyeTrackVR.git
synced 2025-11-04 14:39:42 +08:00
Revert "Cleaning+SafeCrop"
This commit is contained in:
parent
dd23dcf838
commit
ac4447f0ba
@ -1,4 +1,4 @@
|
|||||||
"""
|
'''
|
||||||
------------------------------------------------------------------------------------------------------
|
------------------------------------------------------------------------------------------------------
|
||||||
|
|
||||||
,@@@@@@
|
,@@@@@@
|
||||||
@ -19,8 +19,8 @@
|
|||||||
@@@@@@@@@@@@@@@@@
|
@@@@@@@@@@@@@@@@@
|
||||||
@@@@@@@@@@@@@(
|
@@@@@@@@@@@@@(
|
||||||
|
|
||||||
HSR By: PallasNeko (Optimization Wizard, Contributor), Summer#2406 (Main Algorithm Engineer)
|
HSR By: Sean.Denka (Optimization Wizard, Contributor), Summer#2406 (Main Algorithm Engineer)
|
||||||
RANSAC 3D By: Summer#2406 (Main Algorithm Engineer), Pupil Labs (pye3d), PallasNeko (Optimization)
|
RANSAC 3D By: Summer#2406 (Main Algorithm Engineer), Pupil Labs (pye3d), Sean.Denka (Optimization)
|
||||||
BLOB By: Prohurtz#0001 (Main App Developer)
|
BLOB By: Prohurtz#0001 (Main App Developer)
|
||||||
Algorithm App Implimentations By: Prohurtz#0001, qdot (Inital App Creator)
|
Algorithm App Implimentations By: Prohurtz#0001, qdot (Inital App Creator)
|
||||||
|
|
||||||
@ -28,7 +28,7 @@ Additional Contributors: [Assassin], Summer404NotFound, lorow, ZanzyTHEbar
|
|||||||
|
|
||||||
Copyright (c) 2022 EyeTrackVR <3
|
Copyright (c) 2022 EyeTrackVR <3
|
||||||
------------------------------------------------------------------------------------------------------
|
------------------------------------------------------------------------------------------------------
|
||||||
"""
|
'''
|
||||||
|
|
||||||
from operator import truth
|
from operator import truth
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
@ -46,20 +46,19 @@ import numpy as np
|
|||||||
import cv2
|
import cv2
|
||||||
from enum import Enum
|
from enum import Enum
|
||||||
from one_euro_filter import OneEuroFilter
|
from one_euro_filter import OneEuroFilter
|
||||||
|
|
||||||
if sys.platform.startswith("win"):
|
if sys.platform.startswith("win"):
|
||||||
from winsound import PlaySound, SND_FILENAME, SND_ASYNC
|
from winsound import PlaySound, SND_FILENAME, SND_ASYNC
|
||||||
|
|
||||||
import importlib
|
import importlib
|
||||||
from osc_calibrate_filter import *
|
from osc_calibrate_filter import *
|
||||||
from haar_surround_feature import External_Run_HSF
|
from haar_surround_feature import *
|
||||||
from blob import *
|
from blob import *
|
||||||
from ransac import *
|
from ransac import *
|
||||||
from hsrac import External_Run_HSRACS
|
from hsrac import *
|
||||||
from blink import *
|
from blink import *
|
||||||
|
|
||||||
from intensity_eye_open import *
|
|
||||||
|
|
||||||
|
from intensity_eye_open import *
|
||||||
|
|
||||||
class InformationOrigin(Enum):
|
class InformationOrigin(Enum):
|
||||||
RANSAC = 1
|
RANSAC = 1
|
||||||
@ -68,10 +67,7 @@ class InformationOrigin(Enum):
|
|||||||
HSF = 4
|
HSF = 4
|
||||||
HSRAC = 5
|
HSRAC = 5
|
||||||
|
|
||||||
|
|
||||||
bbb = 0
|
bbb = 0
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class EyeInformation:
|
class EyeInformation:
|
||||||
info_type: InformationOrigin
|
info_type: InformationOrigin
|
||||||
@ -89,7 +85,7 @@ def run_once(f):
|
|||||||
if not wrapper.has_run:
|
if not wrapper.has_run:
|
||||||
wrapper.has_run = True
|
wrapper.has_run = True
|
||||||
return f(*args, **kwargs)
|
return f(*args, **kwargs)
|
||||||
|
|
||||||
wrapper.has_run = False
|
wrapper.has_run = False
|
||||||
return wrapper
|
return wrapper
|
||||||
|
|
||||||
@ -98,35 +94,36 @@ async def delayed_setting_change(setting, value):
|
|||||||
await asyncio.sleep(5)
|
await asyncio.sleep(5)
|
||||||
setting = value
|
setting = value
|
||||||
if sys.platform.startswith("win"):
|
if sys.platform.startswith("win"):
|
||||||
PlaySound("Audio/compleated.wav", SND_FILENAME | SND_ASYNC)
|
PlaySound('Audio/compleated.wav', SND_FILENAME | SND_ASYNC)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class EyeProcessor:
|
class EyeProcessor:
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
config: "EyeTrackCameraConfig",
|
config: "EyeTrackCameraConfig",
|
||||||
settings: "EyeTrackSettingsConfig",
|
settings: "EyeTrackSettingsConfig",
|
||||||
cancellation_event: "threading.Event",
|
cancellation_event: "threading.Event",
|
||||||
capture_event: "threading.Event",
|
capture_event: "threading.Event",
|
||||||
capture_queue_incoming: "queue.Queue",
|
capture_queue_incoming: "queue.Queue",
|
||||||
image_queue_outgoing: "queue.Queue",
|
image_queue_outgoing: "queue.Queue",
|
||||||
eye_id,
|
eye_id,
|
||||||
):
|
):
|
||||||
self.config = config
|
self.config = config
|
||||||
self.settings = settings
|
self.settings = settings
|
||||||
|
|
||||||
# Cross-thread communication management
|
# Cross-thread communication management
|
||||||
self.capture_queue_incoming = capture_queue_incoming
|
self.capture_queue_incoming = capture_queue_incoming
|
||||||
self.image_queue_outgoing = image_queue_outgoing
|
self.image_queue_outgoing = image_queue_outgoing
|
||||||
self.cancellation_event = cancellation_event
|
self.cancellation_event = cancellation_event
|
||||||
self.capture_event = capture_event
|
self.capture_event = capture_event
|
||||||
self.eye_id = eye_id
|
self.eye_id = eye_id
|
||||||
|
|
||||||
# Cross algo state
|
# Cross algo state
|
||||||
self.lkg_projected_sphere = None
|
self.lkg_projected_sphere = None
|
||||||
self.xc = None
|
self.xc = None
|
||||||
self.yc = None
|
self.yc = None
|
||||||
|
|
||||||
# Image state
|
# Image state
|
||||||
self.previous_image = None
|
self.previous_image = None
|
||||||
self.current_image = None
|
self.current_image = None
|
||||||
@ -134,14 +131,14 @@ class EyeProcessor:
|
|||||||
self.current_frame_number = None
|
self.current_frame_number = None
|
||||||
self.current_fps = None
|
self.current_fps = None
|
||||||
self.threshold_image = None
|
self.threshold_image = None
|
||||||
|
|
||||||
# Calibration Values
|
# Calibration Values
|
||||||
self.xoff = 1
|
self.xoff = 1
|
||||||
self.yoff = 1
|
self.yoff = 1
|
||||||
# Keep large in order to recenter correctly
|
# Keep large in order to recenter correctly
|
||||||
self.calibration_frame_counter = None
|
self.calibration_frame_counter = None
|
||||||
self.eyeoffx = 1
|
self.eyeoffx = 1
|
||||||
|
|
||||||
self.xmax = -69420
|
self.xmax = -69420
|
||||||
self.xmin = 69420
|
self.xmin = 69420
|
||||||
self.ymax = -69420
|
self.ymax = -69420
|
||||||
@ -150,41 +147,44 @@ class EyeProcessor:
|
|||||||
self.cccs = False
|
self.cccs = False
|
||||||
self.ts = 10
|
self.ts = 10
|
||||||
self.previous_rotation = self.config.rotation_angle
|
self.previous_rotation = self.config.rotation_angle
|
||||||
|
self.calibration_frame_counter
|
||||||
self.camera_model = None
|
self.camera_model = None
|
||||||
self.detector_3d = None
|
self.detector_3d = None
|
||||||
|
|
||||||
self.camera_model = None
|
self.camera_model = None
|
||||||
self.detector_3d = None
|
self.detector_3d = None
|
||||||
|
|
||||||
self.failed = 0
|
self.failed = 0
|
||||||
|
|
||||||
self.skip_blink_detect = False
|
self.skip_blink_detect = False
|
||||||
|
|
||||||
# blink
|
#blink
|
||||||
self.max_ints = []
|
self.max_ints = []
|
||||||
self.max_int = 0
|
self.max_int = 0
|
||||||
self.min_int = 4000000000000
|
self.min_int = 4000000000000
|
||||||
self.frames = 0
|
self.frames = 0
|
||||||
self.blinkvalue = False
|
self.blinkvalue = False
|
||||||
|
|
||||||
self.prev_x = None
|
self.prev_x = None
|
||||||
self.prev_y = None
|
self.prev_y = None
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
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, min_cutoff=min_cutoff, beta=beta
|
noisy_point,
|
||||||
|
min_cutoff=min_cutoff,
|
||||||
|
beta=beta
|
||||||
)
|
)
|
||||||
|
|
||||||
def output_images_and_update(
|
def output_images_and_update(self, threshold_image, output_information: EyeInformation):
|
||||||
self, threshold_image, output_information: EyeInformation
|
|
||||||
):
|
|
||||||
try:
|
try:
|
||||||
image_stack = np.concatenate(
|
image_stack = np.concatenate(
|
||||||
(
|
(
|
||||||
@ -196,32 +196,29 @@ 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(
|
print('\033[91m[ERROR] Size of frames to display are of unequal sizes.\033[0m')
|
||||||
"\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
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
|
|
||||||
except:
|
except:
|
||||||
# Failure to process frame, reuse previous frame.
|
# Failure to process frame, reuse previous frame.
|
||||||
self.current_image = self.previous_image
|
self.current_image = self.previous_image
|
||||||
print("\033[91m[ERROR] Frame capture issue detected.\033[0m")
|
print("\033[91m[ERROR] Frame capture issue detected.\033[0m")
|
||||||
|
|
||||||
try:
|
try:
|
||||||
# Apply rotation to cropped area. For any rotation area outside of the bounds of the image,
|
# Apply rotation to cropped area. For any rotation area outside of the bounds of the image,
|
||||||
# fill with white.
|
# fill with white.
|
||||||
@ -243,119 +240,96 @@ class EyeProcessor:
|
|||||||
return True
|
return True
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def BLINKM(self):
|
def BLINKM(self):
|
||||||
self.blinkvalue = BLINK(self)
|
self.blinkvalue = BLINK(self)
|
||||||
|
|
||||||
|
|
||||||
def HSRACM(self):
|
def HSRACM(self):
|
||||||
# temporary implementation
|
cx, cy, thresh, gray_frame, uncropframe = External_Run_HSRACS.HSRACS(self)
|
||||||
cx, cy, thresh, gray_frame, uncropframe = External_Run_HSRACS().run(
|
|
||||||
self.current_image_gray
|
|
||||||
)
|
|
||||||
self.current_image_gray = gray_frame
|
self.current_image_gray = gray_frame
|
||||||
if self.prev_x is None:
|
if self.prev_x == None:
|
||||||
self.prev_x = cx
|
self.prev_x = cx
|
||||||
self.prev_y = cy
|
self.prev_y = cy
|
||||||
# print(self.prev_x, self.prev_y, cx, cy)
|
#print(self.prev_x, self.prev_y, cx, cy)
|
||||||
# #filter values with too much movement
|
# #filter values with too much movement
|
||||||
# if (cx - self.prev_x) <= 45 and (cy - self.prev_y) <= 45 :
|
# if (cx - self.prev_x) <= 45 and (cy - self.prev_y) <= 45 :
|
||||||
# self.prev_x = cx
|
# self.prev_x = cx
|
||||||
# self.prev_y = cy
|
# self.prev_y = cy
|
||||||
eyeopen = intense(cx, cy, uncropframe)
|
eyeopen = intense(cx, cy, uncropframe)
|
||||||
out_x, out_y = cal_osc(self, cx, cy)
|
out_x, out_y = cal_osc(self, cx, cy)
|
||||||
|
|
||||||
if cx == 0:
|
if cx == 0:
|
||||||
self.output_images_and_update(
|
self.output_images_and_update(thresh, EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, eyeopen)) #update app
|
||||||
thresh,
|
|
||||||
EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, eyeopen),
|
|
||||||
) # update app
|
|
||||||
else:
|
else:
|
||||||
|
|
||||||
self.output_images_and_update(
|
self.output_images_and_update(thresh, EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, eyeopen))
|
||||||
thresh,
|
# else:
|
||||||
EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, eyeopen),
|
# print("EYE MOVED TOO FAST")
|
||||||
)
|
# self.output_images_and_update(thresh, EyeInformation(InformationOrigin.HSRAC, 0, 0, 0, False))
|
||||||
|
|
||||||
# else:
|
|
||||||
# print("EYE MOVED TOO FAST")
|
|
||||||
# self.output_images_and_update(thresh, EyeInformation(InformationOrigin.HSRAC, 0, 0, 0, False))
|
|
||||||
def HSFM(self):
|
def HSFM(self):
|
||||||
# temporary implementation
|
cx, cy, frame = External_Run_HSF.HSFS(self)
|
||||||
cx, cy, frame = External_Run_HSF().run(self.current_image_gray)
|
|
||||||
eyeopen = intense(cx, cy, self.current_image_gray)
|
eyeopen = intense(cx, cy, self.current_image_gray)
|
||||||
out_x, out_y = cal_osc(self, cx, cy)
|
out_x, out_y = cal_osc(self, cx, cy)
|
||||||
if cx == 0:
|
if cx == 0:
|
||||||
self.output_images_and_update(
|
self.output_images_and_update(frame, EyeInformation(InformationOrigin.HSF, out_x, out_y, 0, eyeopen)) #update app
|
||||||
frame, EyeInformation(InformationOrigin.HSF, out_x, out_y, 0, eyeopen)
|
|
||||||
) # update app
|
|
||||||
else:
|
else:
|
||||||
self.output_images_and_update(
|
self.output_images_and_update(frame, EyeInformation(InformationOrigin.HSF, out_x, out_y, 0, eyeopen))
|
||||||
frame, EyeInformation(InformationOrigin.HSF, out_x, out_y, 0, eyeopen)
|
|
||||||
)
|
|
||||||
|
|
||||||
def RANSAC3DM(self):
|
def RANSAC3DM(self):
|
||||||
cx, cy, thresh = RANSAC3D(self)
|
cx, cy, thresh = RANSAC3D(self)
|
||||||
eyeopen = intense(cx, cy, self.current_image_gray)
|
eyeopen = intense(cx, cy, self.current_image_gray)
|
||||||
out_x, out_y = cal_osc(self, cx, cy)
|
out_x, out_y = cal_osc(self, cx, cy)
|
||||||
if cx == 0:
|
if cx == 0:
|
||||||
self.output_images_and_update(
|
self.output_images_and_update(thresh, EyeInformation(InformationOrigin.RANSAC, out_x, out_y, 0, eyeopen)) #update app
|
||||||
thresh,
|
|
||||||
EyeInformation(InformationOrigin.RANSAC, out_x, out_y, 0, eyeopen),
|
|
||||||
) # update app
|
|
||||||
else:
|
else:
|
||||||
self.output_images_and_update(
|
self.output_images_and_update(thresh, EyeInformation(InformationOrigin.RANSAC, out_x, out_y, 0, eyeopen))
|
||||||
thresh,
|
|
||||||
EyeInformation(InformationOrigin.RANSAC, out_x, out_y, 0, eyeopen),
|
|
||||||
)
|
|
||||||
|
|
||||||
def BLOBM(self):
|
def BLOBM(self):
|
||||||
cx, cy, thresh = BLOB(self)
|
cx, cy, thresh = BLOB(self)
|
||||||
eyeopen = intense(cx, cy, self.current_image_gray)
|
eyeopen = intense(cx, cy, self.current_image_gray)
|
||||||
out_x, out_y = cal_osc(self, cx, cy)
|
out_x, out_y = cal_osc(self, cx, cy)
|
||||||
if cx == 0:
|
if cx == 0:
|
||||||
self.output_images_and_update(
|
self.output_images_and_update(thresh, EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, eyeopen)) #update app
|
||||||
thresh,
|
|
||||||
EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, eyeopen),
|
|
||||||
) # update app
|
|
||||||
else:
|
else:
|
||||||
self.output_images_and_update(
|
self.output_images_and_update(thresh, EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, eyeopen))
|
||||||
thresh,
|
|
||||||
EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, eyeopen),
|
|
||||||
)
|
|
||||||
|
|
||||||
def ALGOSELECT(self):
|
|
||||||
|
|
||||||
if self.failed == 0 and self.firstalgo is not None:
|
|
||||||
|
|
||||||
|
def ALGOSELECT(self):
|
||||||
|
|
||||||
|
if self.failed == 0 and self.firstalgo != None:
|
||||||
self.firstalgo()
|
self.firstalgo()
|
||||||
else:
|
else:
|
||||||
self.failed = self.failed + 1
|
self.failed = self.failed + 1
|
||||||
|
|
||||||
if (
|
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.failed == 1 and self.secondalgo is not 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
|
||||||
|
|
||||||
if self.failed == 2 and self.thirdalgo is not None:
|
if self.failed == 2 and self.thirdalgo != None:
|
||||||
self.thirdalgo()
|
self.thirdalgo()
|
||||||
else:
|
else:
|
||||||
self.failed = self.failed + 1
|
self.failed = self.failed + 1
|
||||||
|
|
||||||
if self.failed == 3 and self.fourthalgo is not None:
|
if self.failed == 3 and self.fourthalgo != None:
|
||||||
self.fourthalgo()
|
self.fourthalgo()
|
||||||
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):
|
||||||
|
|
||||||
self.firstalgo = None
|
self.firstalgo = None
|
||||||
self.secondalgo = None
|
self.secondalgo = None
|
||||||
self.thirdalgo = None
|
self.thirdalgo = None
|
||||||
self.fourthalgo = None
|
self.fourthalgo = None
|
||||||
# set algo priorities
|
#set algo priorities
|
||||||
|
|
||||||
if (self.settings.gui_HSF and self.settings.gui_HSFP == 1): # I feel like this is super innefficient though it only runs at startup and no solution is coming to me atm
|
if self.settings.gui_HSF and self.settings.gui_HSFP == 1: #I feel like this is super innefficient though it only runs at startup and no solution is coming to me atm
|
||||||
self.firstalgo = self.HSFM
|
self.firstalgo = self.HSFM
|
||||||
elif self.settings.gui_HSF and self.settings.gui_HSFP == 2:
|
elif self.settings.gui_HSF and self.settings.gui_HSFP == 2:
|
||||||
self.secondalgo = self.HSFM
|
self.secondalgo = self.HSFM
|
||||||
@ -363,7 +337,7 @@ class EyeProcessor:
|
|||||||
self.thirdalgo = self.HSFM
|
self.thirdalgo = self.HSFM
|
||||||
elif self.settings.gui_HSF and self.settings.gui_HSFP == 4:
|
elif self.settings.gui_HSF and self.settings.gui_HSFP == 4:
|
||||||
self.fourthalgo = self.HSFM
|
self.fourthalgo = self.HSFM
|
||||||
|
|
||||||
if self.settings.gui_RANSAC3D and self.settings.gui_RANSAC3DP == 1:
|
if self.settings.gui_RANSAC3D and self.settings.gui_RANSAC3DP == 1:
|
||||||
self.firstalgo = self.RANSAC3DM
|
self.firstalgo = self.RANSAC3DM
|
||||||
elif self.settings.gui_RANSAC3D and self.settings.gui_RANSAC3DP == 2:
|
elif self.settings.gui_RANSAC3D and self.settings.gui_RANSAC3DP == 2:
|
||||||
@ -372,8 +346,8 @@ class EyeProcessor:
|
|||||||
self.thirdalgo = self.RANSAC3DM
|
self.thirdalgo = self.RANSAC3DM
|
||||||
elif self.settings.gui_RANSAC3D and self.settings.gui_RANSAC3DP == 4:
|
elif self.settings.gui_RANSAC3D and self.settings.gui_RANSAC3DP == 4:
|
||||||
self.fourthalgo = self.RANSAC3DM
|
self.fourthalgo = self.RANSAC3DM
|
||||||
|
|
||||||
if self.settings.gui_HSRAC and self.settings.gui_HSRACP == 1:
|
if self.settings.gui_HSRAC == True and self.settings.gui_HSRACP == 1:
|
||||||
self.firstalgo = self.HSRACM
|
self.firstalgo = self.HSRACM
|
||||||
elif self.settings.gui_HSRAC and self.settings.gui_HSRACP == 2:
|
elif self.settings.gui_HSRAC and self.settings.gui_HSRACP == 2:
|
||||||
self.secondalgo = self.HSRACM
|
self.secondalgo = self.HSRACM
|
||||||
@ -381,7 +355,7 @@ class EyeProcessor:
|
|||||||
self.thirdalgo = self.HSRACM
|
self.thirdalgo = self.HSRACM
|
||||||
elif self.settings.gui_HSRAC and self.settings.gui_HSRACP == 4:
|
elif self.settings.gui_HSRAC and self.settings.gui_HSRACP == 4:
|
||||||
self.fourthalgo = self.HSRACM
|
self.fourthalgo = self.HSRACM
|
||||||
|
|
||||||
if self.settings.gui_BLOB and self.settings.gui_BLOBP == 1:
|
if self.settings.gui_BLOB and self.settings.gui_BLOBP == 1:
|
||||||
self.firstalgo = self.BLOBM
|
self.firstalgo = self.BLOBM
|
||||||
elif self.settings.gui_BLOB and self.settings.gui_BLOBP == 2:
|
elif self.settings.gui_BLOB and self.settings.gui_BLOBP == 2:
|
||||||
@ -390,31 +364,31 @@ class EyeProcessor:
|
|||||||
self.thirdalgo = self.BLOBM
|
self.thirdalgo = self.BLOBM
|
||||||
elif self.settings.gui_BLOB and self.settings.gui_BLOBP == 4:
|
elif self.settings.gui_BLOB and self.settings.gui_BLOBP == 4:
|
||||||
self.fourthalgo = self.BLOBM
|
self.fourthalgo = self.BLOBM
|
||||||
|
|
||||||
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.
|
||||||
if self.cancellation_event.wait(0.1):
|
if self.cancellation_event.wait(0.1):
|
||||||
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 (
|
if (self.camera_model is None
|
||||||
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,
|
||||||
)
|
)
|
||||||
):
|
):
|
||||||
self.camera_model = CameraModel(
|
self.camera_model = CameraModel(
|
||||||
focal_length=self.config.focal_length,
|
focal_length=self.config.focal_length,
|
||||||
@ -423,7 +397,7 @@ class EyeProcessor:
|
|||||||
self.detector_3d = Detector3D(
|
self.detector_3d = Detector3D(
|
||||||
camera=self.camera_model, long_term_mode=DetectorMode.blocking
|
camera=self.camera_model, long_term_mode=DetectorMode.blocking
|
||||||
)
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
if self.capture_queue_incoming.empty():
|
if self.capture_queue_incoming.empty():
|
||||||
self.capture_event.set()
|
self.capture_event.set()
|
||||||
@ -439,35 +413,41 @@ 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_clean = self.current_image_gray.copy() #copy this frame to have a clean image for blink algo
|
||||||
self.current_image_gray.copy()
|
# print(self.settings.gui_RANSAC3D)
|
||||||
) # copy this frame to have a clean image for blink algo
|
|
||||||
# print(self.settings.gui_RANSAC3D)
|
# BLINK(self)
|
||||||
|
|
||||||
|
# cx, cy, thresh = HSRAC(self)
|
||||||
|
# out_x, out_y = cal_osc(self, cx, cy)
|
||||||
|
# if cx == 0:
|
||||||
|
# self.output_images_and_update(thresh, EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, True)) #update app
|
||||||
|
# else:
|
||||||
|
# self.output_images_and_update(thresh, EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, self.blinkvalue))
|
||||||
|
|
||||||
# BLINK(self)
|
|
||||||
|
# cx, cy, thresh = RANSAC3D(self)
|
||||||
|
# out_x, out_y = cal_osc(self, cx, cy)
|
||||||
|
# self.output_images_and_update(thresh, EyeInformation(InformationOrigin.RANSAC, out_x, out_y, 0, False)) #update app
|
||||||
|
|
||||||
|
|
||||||
|
# cx, cy, larger_threshold = BLOB(self)
|
||||||
|
# out_x, out_y = cal_osc(self, cx, cy)
|
||||||
|
# self.output_images_and_update(larger_threshold, EyeInformation(InformationOrigin.BLOB, out_x, out_y, 0, False)) #update app
|
||||||
|
|
||||||
|
#center_x, center_y, frame = HSF(self) #run algo
|
||||||
|
#out_x, out_y = cal_osc(self, center_x, center_y) #filter and calibrate
|
||||||
|
#self.output_images_and_update(frame, EyeInformation(InformationOrigin.HSF, out_x, out_y, 0, False)) #update app
|
||||||
|
|
||||||
# cx, cy, thresh = HSRAC(self)
|
self.ALGOSELECT() #run our algos in priority order set in settings
|
||||||
# out_x, out_y = cal_osc(self, cx, cy)
|
|
||||||
# if cx == 0:
|
|
||||||
# self.output_images_and_update(thresh, EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, True)) #update app
|
|
||||||
# else:
|
|
||||||
# self.output_images_and_update(thresh, EyeInformation(InformationOrigin.HSRAC, out_x, out_y, 0, self.blinkvalue))
|
|
||||||
|
|
||||||
# cx, cy, thresh = RANSAC3D(self)
|
|
||||||
# out_x, out_y = cal_osc(self, cx, cy)
|
|
||||||
# self.output_images_and_update(thresh, EyeInformation(InformationOrigin.RANSAC, out_x, out_y, 0, False)) #update app
|
|
||||||
|
|
||||||
# cx, cy, larger_threshold = BLOB(self)
|
|
||||||
# out_x, out_y = cal_osc(self, cx, cy)
|
|
||||||
# self.output_images_and_update(larger_threshold, EyeInformation(InformationOrigin.BLOB, out_x, out_y, 0, False)) #update app
|
|
||||||
|
|
||||||
# center_x, center_y, frame = HSF(self) #run algo
|
|
||||||
# out_x, out_y = cal_osc(self, center_x, center_y) #filter and calibrate
|
|
||||||
# self.output_images_and_update(frame, EyeInformation(InformationOrigin.HSF, out_x, out_y, 0, False)) #update app
|
|
||||||
|
|
||||||
self.ALGOSELECT() # run our algos in priority order set in settings
|
|
||||||
self.BLINKM()
|
self.BLINKM()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@ -19,7 +19,7 @@
|
|||||||
@@@@@@@@@@@@@@@@@
|
@@@@@@@@@@@@@@@@@
|
||||||
@@@@@@@@@@@@@(
|
@@@@@@@@@@@@@(
|
||||||
|
|
||||||
RANSAC 3D By: Summer#2406 (Main Algorithm Engineer), Pupil Labs (pye3d), PallasNeko (Optimization)
|
RANSAC 3D By: Summer#2406 (Main Algorithm Engineer), Pupil Labs (pye3d), Sean.Denka (Optimization)
|
||||||
Algorithm App Implimentations By: Prohurtz#0001, qdot (Inital App Creator)
|
Algorithm App Implimentations By: Prohurtz#0001, qdot (Inital App Creator)
|
||||||
|
|
||||||
Copyright (c) 2022 EyeTrackVR <3
|
Copyright (c) 2022 EyeTrackVR <3
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user