瀏覽代碼

Update file_utils.py

add speed_change function
刘悦 1 年之前
父節點
當前提交
cf43100f66
共有 1 個文件被更改,包括 12 次插入0 次删除
  1. 12 0
      cosyvoice/utils/file_utils.py

+ 12 - 0
cosyvoice/utils/file_utils.py

@@ -39,3 +39,15 @@ def load_wav(wav, target_sr):
         assert sample_rate > target_sr, 'wav sample rate {} must be greater than {}'.format(sample_rate, target_sr)
         speech = torchaudio.transforms.Resample(orig_freq=sample_rate, new_freq=target_sr)(speech)
     return speech
+
+def speed_change(waveform,sample_rate,speed_factor:str):
+    effects = [
+    ["tempo",speed_factor],  # speed_factor
+    ["rate", f"{sample_rate}"]  
+    ]
+    augmented_waveform, new_sample_rate = torchaudio.sox_effects.apply_effects_tensor(
+        waveform, 
+        sample_rate, 
+        effects
+    )
+    return augmented_waveform, new_sample_rate