mirror of
https://github.com/EyeTrackVR/EyeTrackVR.git
synced 2025-11-04 14:39:42 +08:00
fix BLOB remove debug prints
This commit is contained in:
parent
1c7c6d964b
commit
2465df6697
@ -36,47 +36,10 @@ class EyeId(IntEnum):
|
|||||||
BOTH = 2
|
BOTH = 2
|
||||||
SETTINGS = 3
|
SETTINGS = 3
|
||||||
|
|
||||||
def circle_crop(self):
|
|
||||||
global cct
|
|
||||||
print(cct)
|
|
||||||
if cct == 0:
|
|
||||||
try:
|
|
||||||
ht, wd = self.current_image_gray.shape[:2]
|
|
||||||
radius = int(float(self.lkg_projected_sphere["axes"][0]))
|
|
||||||
self.xc = int(float(self.lkg_projected_sphere["center"][0]))
|
|
||||||
self.yc = int(float(self.lkg_projected_sphere["center"][1]))
|
|
||||||
if radius < 10: #minimum size
|
|
||||||
radius = 10
|
|
||||||
# draw filled circle in white on black background as mask
|
|
||||||
mask = np.zeros((ht, wd), dtype=np.uint8)
|
|
||||||
mask = cv2.circle(mask, (self.xc, self.yc), radius, 255, -1)
|
|
||||||
# create white colored background
|
|
||||||
color = np.full_like(self.current_image_gray, (255))
|
|
||||||
# apply mask to image
|
|
||||||
masked_img = cv2.bitwise_and(self.current_image_gray, self.current_image_gray, mask=mask)
|
|
||||||
# apply inverse mask to colored image
|
|
||||||
masked_color = cv2.bitwise_and(color, color, mask=255 - mask)
|
|
||||||
# combine the two masked images
|
|
||||||
self.current_image_gray = cv2.add(masked_img, masked_color)
|
|
||||||
return self.current_image_gray
|
|
||||||
except:
|
|
||||||
return self.current_image_gray
|
|
||||||
pass
|
|
||||||
else:
|
|
||||||
cct = cct - 1
|
|
||||||
return self.current_image_gray
|
|
||||||
def BLOB(self):
|
def BLOB(self):
|
||||||
global cct
|
global cct
|
||||||
# define circle
|
# define circle
|
||||||
if self.eye_id in [EyeId.LEFT] and self.settings.gui_circular_crop_left:
|
|
||||||
self.current_image_gray = circle_crop(self)
|
|
||||||
else:
|
|
||||||
pass
|
|
||||||
|
|
||||||
if self.eye_id in [EyeId.RIGHT] and self.settings.gui_circular_crop_right:
|
|
||||||
self.current_image_gray = circle_crop(self)
|
|
||||||
else:
|
|
||||||
pass
|
|
||||||
_, larger_threshold = cv2.threshold(self.current_image_gray, int(self.settings.gui_threshold), 255,
|
_, larger_threshold = cv2.threshold(self.current_image_gray, int(self.settings.gui_threshold), 255,
|
||||||
cv2.THRESH_BINARY)
|
cv2.THRESH_BINARY)
|
||||||
|
|
||||||
@ -114,10 +77,10 @@ def BLOB(self):
|
|||||||
|
|
||||||
#out_x, out_y = cal_osc(self, cx, cy) #filter and calibrate values
|
#out_x, out_y = cal_osc(self, cx, cy) #filter and calibrate values
|
||||||
|
|
||||||
print("S:KLJGHLDKIJGHLKS")
|
|
||||||
self.failed = 0
|
self.failed = 0
|
||||||
return cx, cy, larger_threshold
|
return cx, cy, larger_threshold
|
||||||
print("S:KLJGHLDKIJGHLKS")
|
|
||||||
self.failed = self.failed + 1
|
self.failed = self.failed + 1
|
||||||
return 0, 0, larger_threshold
|
return 0, 0, larger_threshold
|
||||||
|
|
||||||
@ -334,8 +334,14 @@ class EyeProcessor:
|
|||||||
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:
|
||||||
print("LSKDGFHL")
|
self.current_image_gray, self.cct = circle_crop(self.current_image_gray, self.xc, self.yc, self.cc_radius, self.cct)
|
||||||
|
else:
|
||||||
|
pass
|
||||||
|
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)
|
||||||
|
else:
|
||||||
|
pass
|
||||||
self.rawx, self.rawy, self.thresh = BLOB(self)
|
self.rawx, self.rawy, self.thresh = BLOB(self)
|
||||||
|
|
||||||
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)
|
||||||
@ -362,6 +368,8 @@ class EyeProcessor:
|
|||||||
|
|
||||||
if self.failed == 3 and self.fourthalgo != None:
|
if self.failed == 3 and self.fourthalgo != None:
|
||||||
self.fourthalgo()
|
self.fourthalgo()
|
||||||
|
else:
|
||||||
|
self.failed = self.failed + 1
|
||||||
|
|
||||||
if self.failed == 4 and self.fithalgo != None:
|
if self.failed == 4 and self.fithalgo != None:
|
||||||
self.fithalgo()
|
self.fithalgo()
|
||||||
|
|||||||
@ -263,7 +263,6 @@ def main():
|
|||||||
window[ALGO_SETTINGS_NAME].update(visible=False)
|
window[ALGO_SETTINGS_NAME].update(visible=False)
|
||||||
config.eye_display_id = EyeId.SETTINGS
|
config.eye_display_id = EyeId.SETTINGS
|
||||||
config.save()
|
config.save()
|
||||||
print("EE")
|
|
||||||
|
|
||||||
|
|
||||||
elif values[ALGO_SETTINGS_RADIO_NAME] and config.eye_display_id != EyeId.ALGOSETTINGS:
|
elif values[ALGO_SETTINGS_RADIO_NAME] and config.eye_display_id != EyeId.ALGOSETTINGS:
|
||||||
@ -277,7 +276,6 @@ def main():
|
|||||||
window[ALGO_SETTINGS_NAME].update(visible=True)
|
window[ALGO_SETTINGS_NAME].update(visible=True)
|
||||||
config.eye_display_id = EyeId.ALGOSETTINGS
|
config.eye_display_id = EyeId.ALGOSETTINGS
|
||||||
config.save()
|
config.save()
|
||||||
print("Easljghk")
|
|
||||||
|
|
||||||
# Otherwise, render all
|
# Otherwise, render all
|
||||||
for eye in eyes:
|
for eye in eyes:
|
||||||
|
|||||||
@ -228,7 +228,7 @@ class IntensityBasedOpeness:
|
|||||||
else:
|
else:
|
||||||
self.filterlist.pop()
|
self.filterlist.pop()
|
||||||
self.filterlist.append(intensity)
|
self.filterlist.append(intensity)
|
||||||
print(len(str(intensity)))
|
|
||||||
if len(str(intensity)) >= 8: #filter abnormally high values
|
if len(str(intensity)) >= 8: #filter abnormally high values
|
||||||
print('filter, assume blink')
|
print('filter, assume blink')
|
||||||
intensity = self.maxval
|
intensity = self.maxval
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user