diff --git a/EyeTrackApp/leap.py b/EyeTrackApp/leap.py index d650610..4b6eb85 100644 --- a/EyeTrackApp/leap.py +++ b/EyeTrackApp/leap.py @@ -16,6 +16,7 @@ os.environ["OMP_NUM_THREADS"] = "1" frames = 0 models = Path("Models") + def run_model(input_queue, output_queue, session): while True: frame = input_queue.get() @@ -32,12 +33,14 @@ def run_model(input_queue, output_queue, session): pre_landmark = np.reshape(pre_landmark, (-1, 2)) output_queue.put((frame, pre_landmark)) + def run_onnx_model(queues, session, frame): for queue in queues: if not queue.full(): queue.put(frame) break + class LEAP_C: def __init__(self): self.last_lid = None @@ -74,8 +77,7 @@ class LEAP_C: self.total_velocity_old = 0 self.old_per = 0.0 self.delta_per_neg = 0.0 - self.ort_session1 = onnxruntime.InferenceSession(self.model_path, opts, providers=['CUDAExecutionProvider', 'CPUExecutionProvider']) - + self.ort_session1 = onnxruntime.InferenceSession(self.model_path, opts, providers=["CPUExecutionProvider"]) for i in range(self.num_threads): thread = threading.Thread( @@ -145,6 +147,7 @@ class LEAP_C: imgvis = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY) return imgvis, 0, 0, 0 + class External_Run_LEAP: def __init__(self): self.algo = LEAP_C() @@ -153,4 +156,4 @@ class External_Run_LEAP: self.algo.current_image_gray = current_image_gray self.algo.current_image_gray_clean = current_image_gray_clean img, x, y, per = self.algo.leap_run() - return img, x, y, per \ No newline at end of file + return img, x, y, per