coqui-ai
/
TTS
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix the bug for emptly audio files

pull/1431/head
Edresson Casanova 2022-03-22 14:53:33 -03:00
parent ea53d6feb3
commit 0ae1e0248c
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
TTS/utils/vad.py
View File

@ -18,7 +18,7 @@ def resample_wav(wav, sr, new_sr):
def map_timestamps_to_new_sr(vad_sr, new_sr, timestamps, just_begging_end=False): def map_timestamps_to_new_sr(vad_sr, new_sr, timestamps, just_begging_end=False):
factor = new_sr / vad_sr factor = new_sr / vad_sr
new_timestamps = [] new_timestamps = []
if just_begging_end: if just_begging_end and timestamps:
# get just the start and end timestamps # get just the start and end timestamps
new_dict = {'start': int(timestamps[0]['start']*factor), 'end': int(timestamps[-1]['end']*factor)} new_dict = {'start': int(timestamps[0]['start']*factor), 'end': int(timestamps[-1]['end']*factor)}
new_timestamps.append(new_dict) new_timestamps.append(new_dict)
@ -64,8 +64,12 @@ def remove_silence(model_and_utils, audio_path, out_path, vad_sample_rate=8000,
# map the current speech_timestamps to the sample rate of the ground truth audio # map the current speech_timestamps to the sample rate of the ground truth audio
new_speech_timestamps = map_timestamps_to_new_sr(vad_sample_rate, gt_sample_rate, speech_timestamps, trim_just_beginning_and_end) new_speech_timestamps = map_timestamps_to_new_sr(vad_sample_rate, gt_sample_rate, speech_timestamps, trim_just_beginning_and_end)
# save audio # if have speech timestamps else save the wav
save_audio(out_path, if new_speech_timestamps:
collect_chunks(new_speech_timestamps, wav), sampling_rate=gt_sample_rate) wav = collect_chunks(new_speech_timestamps, wav)
else:
print(f"> The file {audio_path} probably does not have speech please check it !!")
# save audio
save_audio(out_path, wav, sampling_rate=gt_sample_rate)
return out_path return out_path