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

add temp commit

Browse Source
This commit is contained in:
brige 2024-10-30 20:07:04 +08:00
parent cb6dd54497
commit dc5068b237
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
human/audio_inference_handler.py
View File

@ -65,6 +65,7 @@ class AudioInferenceHandler(AudioHandler):
except queue.Empty: except queue.Empty:
continue continue
print('origin mel_batch:', len(mel_batch))
is_all_silence = True is_all_silence = True
audio_frames = [] audio_frames = []
for _ in range(batch_size * 2): for _ in range(batch_size * 2):

4
human/audio_mal_handler.py
View File

@ -58,7 +58,7 @@ class AudioMalHandler(AudioHandler):
# print(mel.shape[0],mel.shape,len(mel[0]),len(self.frames)) # print(mel.shape[0],mel.shape,len(mel[0]),len(self.frames))
# cut off stride # cut off stride
left = max(0, self._context.stride_left_size * 80 / 50) left = max(0, self._context.stride_left_size * 80 / 50)
# right = min(len(mel[0]), len(mel[0]) - self._context.stride_right_size * 80 / 50) right = min(len(mel[0]), len(mel[0]) - self._context.stride_right_size * 80 / 50)
mel_idx_multiplier = 80. * 2 / self._context.fps mel_idx_multiplier = 80. * 2 / self._context.fps
mel_step_size = 16 mel_step_size = 16
i = 0 i = 0
@ -67,8 +67,10 @@ class AudioMalHandler(AudioHandler):
start_idx = int(left + i * mel_idx_multiplier) start_idx = int(left + i * mel_idx_multiplier)
# print(start_idx) # print(start_idx)
if start_idx + mel_step_size > len(mel[0]): if start_idx + mel_step_size > len(mel[0]):
print("AudioMalHandler start_idx", start_idx)
mel_chunks.append(mel[:, len(mel[0]) - mel_step_size:]) mel_chunks.append(mel[:, len(mel[0]) - mel_step_size:])
else: else:
print("AudioMalHandler start_idx222", start_idx + mel_step_size - start_idx)
mel_chunks.append(mel[:, start_idx: start_idx + mel_step_size]) mel_chunks.append(mel[:, start_idx: start_idx + mel_step_size])
i += 1 i += 1
self.on_next_handle(mel_chunks, 1) self.on_next_handle(mel_chunks, 1)