mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve naming

Browse Source
This commit is contained in:
henryruhs 2024-04-02 19:40:36 +02:00
parent efd309acf6
commit a44ff26e39
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
facefusion/voice_extractor.py
View File

@ -65,13 +65,13 @@ def create_static_hanning_window(filter_size : int) -> Any:
def batch_extract_voice(audio : Audio, chunk_size : int, overlap_size : float) -> Audio:
step_size = int(chunk_size * (1 - overlap_size))
audio_total = numpy.zeros((audio.shape[0], 2)).astype(numpy.float32)
audio_count = numpy.zeros((audio.shape[0], 2)).astype(numpy.float32)
chunk_total = numpy.zeros((audio.shape[0], 2)).astype(numpy.float32)
for start in range(0, audio.shape[0], step_size):
end = min(start + chunk_size, audio.shape[0])
audio_total[start:end, ...] += extract_voice(audio[start:end, ...])
audio_count[start:end, ...] += 1
audio = audio_total / audio_count
chunk_total[start:end, ...] += 1
audio = audio_total / chunk_total
return audio