modify run human

This commit is contained in:
brige 2024-09-12 19:17:34 +08:00
parent 90ccaa222b
commit c2140344ce

View File

@ -158,6 +158,7 @@ class Human:
mp.Process(target=inference, args=(self.render_event, self._batch_size, self._face_image_paths,
self._feat_queue, self._output_queue, self._res_frame_queue,
)).start()
self.render_event.set()
def get_fps(self):
return self._fps
@ -175,7 +176,7 @@ class Human:
return self._stride_right_size
def on_destroy(self):
self.render_event.set()
self.render_event.clear()
self._chunk_2_mal.stop()
if self._tts is not None:
self._tts.stop()
@ -208,6 +209,7 @@ class Human:
try:
img, aud = self._res_frame_queue.get(block=True, timeout=.3)
except queue.Empty:
print('queue.Empty:')
return None
return img