diff --git a/human/audio_mal_handler.py b/human/audio_mal_handler.py index bb78507..43104cf 100644 --- a/human/audio_mal_handler.py +++ b/human/audio_mal_handler.py @@ -79,7 +79,6 @@ class AudioMalHandler(AudioHandler): # print(mel.shape[0],mel.shape,len(mel[0]),len(self.frames)) # cut off stride left = max(0, self._context.stride_left_size * 80 / self._context.fps) - right = min(len(mel[0]), len(mel[0]) - self._context.stride_right_size * 80 / 50) mel_idx_multiplier = 80. * 2 / self._context.fps mel_step_size = 16 i = 0 diff --git a/human/human_context.py b/human/human_context.py index 585803f..6d55e8e 100644 --- a/human/human_context.py +++ b/human/human_context.py @@ -18,11 +18,10 @@ class HumanContext: def __init__(self): self._fps = 50 # 20 ms per frame self._image_size = 288 - self._batch_size = 16 + self._batch_size = 6 self._sample_rate = 16000 - self._stride_left_size = 10 - self._stride_right_size = 10 - self._render_batch = 5 + self._stride_left_size = 4 + self._stride_right_size = 4 self._asr = None self._nlp = None @@ -62,10 +61,6 @@ class HumanContext: def image_size(self): return self._image_size - @property - def render_batch(self): - return self._render_batch - @property def device(self): return self._device diff --git a/human/human_render.py b/human/human_render.py index 0c9a25c..efe3a63 100644 --- a/human/human_render.py +++ b/human/human_render.py @@ -52,11 +52,11 @@ class HumanRender(AudioHandler): if self._render_status is RenderStatus.E_Empty: continue real_delay = delay - use - print(f'send voice {use}') + # print(f'send voice {use}') if real_delay > 0: time.sleep(real_delay) - else: - print(f'send voice {real_delay}') + # else: + # print(f'send voice {real_delay}') logging.info('human render exit') diff --git a/ui/ipc_render.py b/ui/ipc_render.py index a15229e..44035dd 100644 --- a/ui/ipc_render.py +++ b/ui/ipc_render.py @@ -49,17 +49,23 @@ class IpcRender(HumanRender): def _on_reader_callback(self, data_str, size): data_str = data_str.decode('utf-8') - print(f'on_reader_callback: {data_str}, size:{size}') + # print(f'on_reader_callback: {data_str}, size:{size}') if 'quit' == data_str: self._context.stop() elif 'heartbeat' == data_str: pass elif 'full' == data_str: - self._render_status = RenderStatus.E_Full + if self._render_status != RenderStatus.E_Full: + logger.info(f'change to E_Full status') + self._render_status = RenderStatus.E_Full elif 'empty' == data_str: - self._render_status = RenderStatus.E_Empty + if self._render_status != RenderStatus.E_Empty: + logger.info(f'change to E_Full status') + self._render_status = RenderStatus.E_Empty elif 'normal' == data_str: - self._render_status = RenderStatus.E_Normal + if self._render_status != RenderStatus.E_Normal: + logger.info(f'change to E_Normal status') + self._render_status = RenderStatus.E_Normal def run(self): self._ipc.set_reader_callback(self._on_reader_callback)