mirror of
https://github.com/EyeTrackVR/EyeTrackVR.git
synced 2025-11-04 14:39:42 +08:00
fix: remove cuda
This commit is contained in:
parent
71ddd8ceba
commit
a291b2fc4f
@ -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
|
||||
return img, x, y, per
|
||||
|
||||
Loading…
Reference in New Issue
Block a user