copilot/OpenSeeFace
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added camera reinitialization.

Browse source
This commit is contained in:
Emiliana 2020-08-10 22:09:17 +02:00
parent 8dbc573830
commit 2fc0f52385
2 changed files with 29 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tracker.py
View file

@ -136,7 +136,10 @@ def worker_thread(session, frame, input, crop_info, queue, input_name, idx, trac
lms = np.array(output[1]) * np.array([crop_info[3], crop_info[2], 1.]) + np.array([crop_info[1], crop_info[0], 0.])
conf = output[0]
if conf > tracker.threshold:
eye_state = tracker.get_eye_state(frame, lms, single=True)
try:
eye_state = tracker.get_eye_state(frame, lms, single=True)
except:
eye_state = [(1.0, 0.0, 0.0, 0.0), (1.0, 0.0, 0.0, 0.0)]
queue.put((session, conf, (lms, eye_state), crop_info, idx))
else:
queue.put((session,))
@ -992,7 +995,10 @@ class Tracker():
lms = np.array(output[1]) * np.array([crop_info[0][3], crop_info[0][2], 1.]) + np.array([crop_info[0][1], crop_info[0][0], 0.])
conf = output[0]
if conf > self.threshold:
eye_state = self.get_eye_state(frame, lms)
try:
eye_state = self.get_eye_state(frame, lms)
except:
eye_state = [(1.0, 0.0, 0.0, 0.0), (1.0, 0.0, 0.0, 0.0)]
outputs[crop_info[0]] = (conf, (lms, eye_state), 0)
else:
started = 0