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

modify audio frame output

Browse Source
This commit is contained in:
brige 2024-09-14 14:21:38 +08:00
parent 504738fb08
commit 9250cef407
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
Human.py
View File

@ -90,7 +90,7 @@ def inference(render_event, batch_size, face_images_path, audio_feat_queue, audi
audio_frames = [] audio_frames = []
is_all_silence = True is_all_silence = True
for _ in range(batch_size * 2): for _ in range(batch_size * 2):
frame, type = audio_feat_queue.get() # is erro frame, type = audio_out_queue.get()
audio_frames.append((frame, type)) audio_frames.append((frame, type))
if type == 0: if type == 0:
@ -205,6 +205,9 @@ class Human:
self._feat_queue.put(mel_chunks) self._feat_queue.put(mel_chunks)
print("22") print("22")
def push_audio_frames(self, chunk, type_):
self._output_queue.put((chunk, type_))
def render(self): def render(self):
try: try:
img, aud = self._res_frame_queue.get(block=True, timeout=.3) img, aud = self._res_frame_queue.get(block=True, timeout=.3)

3
tts/Chunk2Mal.py
View File

@ -21,8 +21,9 @@ class Chunk2Mal:
logging.info('chunk2mal run') logging.info('chunk2mal run')
while not self._exit_event.is_set(): while not self._exit_event.is_set():
try: try:
chunk, type = self.pull_chunk() chunk, type_ = self.pull_chunk()
self._chunks.append(chunk) self._chunks.append(chunk)
self._human.push_audio_frame((chunk, type_))
print("1") print("1")
except queue.Empty: except queue.Empty:
continue continue