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
|
frames = 0
|
||||||
models = Path("Models")
|
models = Path("Models")
|
||||||
|
|
||||||
|
|
||||||
def run_model(input_queue, output_queue, session):
|
def run_model(input_queue, output_queue, session):
|
||||||
while True:
|
while True:
|
||||||
frame = input_queue.get()
|
frame = input_queue.get()
|
||||||
@ -32,12 +33,14 @@ def run_model(input_queue, output_queue, session):
|
|||||||
pre_landmark = np.reshape(pre_landmark, (-1, 2))
|
pre_landmark = np.reshape(pre_landmark, (-1, 2))
|
||||||
output_queue.put((frame, pre_landmark))
|
output_queue.put((frame, pre_landmark))
|
||||||
|
|
||||||
|
|
||||||
def run_onnx_model(queues, session, frame):
|
def run_onnx_model(queues, session, frame):
|
||||||
for queue in queues:
|
for queue in queues:
|
||||||
if not queue.full():
|
if not queue.full():
|
||||||
queue.put(frame)
|
queue.put(frame)
|
||||||
break
|
break
|
||||||
|
|
||||||
|
|
||||||
class LEAP_C:
|
class LEAP_C:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.last_lid = None
|
self.last_lid = None
|
||||||
@ -74,8 +77,7 @@ class LEAP_C:
|
|||||||
self.total_velocity_old = 0
|
self.total_velocity_old = 0
|
||||||
self.old_per = 0.0
|
self.old_per = 0.0
|
||||||
self.delta_per_neg = 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):
|
for i in range(self.num_threads):
|
||||||
thread = threading.Thread(
|
thread = threading.Thread(
|
||||||
@ -145,6 +147,7 @@ class LEAP_C:
|
|||||||
imgvis = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
|
imgvis = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
|
||||||
return imgvis, 0, 0, 0
|
return imgvis, 0, 0, 0
|
||||||
|
|
||||||
|
|
||||||
class External_Run_LEAP:
|
class External_Run_LEAP:
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self.algo = LEAP_C()
|
self.algo = LEAP_C()
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user