brige/human
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases 2 Wiki Activity

modify run human

Browse Source
This commit is contained in:
brige 2024-09-12 19:17:34 +08:00
parent 90ccaa222b
commit c2140344ce
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Human.py
View File

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