Mirror lj1995/VoiceConversionWebUI @ b2c8cae96e3b — trainset_preprocess_pipeline.py

trainset_preprocess_pipeline.py

@@ -0,0 +1,63 @@
+import numpy as np,ffmpeg,os,traceback
+from slicer import Slicer
+slicer = Slicer(
+    sr=40000,
+    db_threshold=-32,
+    min_length=800,
+    win_l=400,
+    win_s=20,
+    max_silence_kept=150
+)
+
+
+
+
+def p0_load_audio(file, sr):#str-ing
+    try:
+        out, _ = (
+            ffmpeg.input(file, threads=0)
+            .output("-", format="s16le", acodec="pcm_s16le", ac=1, ar=sr)
+            .run(cmd=["ffmpeg", "-nostdin"], capture_stdout=True, capture_stderr=True)
+        )
+    except ffmpeg.Error as e:
+        raise RuntimeError(f"Failed to load audio: {e.stderr.decode()}") from e
+    return np.frombuffer(out, np.int16).flatten().astype(np.float32) / 32768.0
+
+def p1_trim_audio(slicer,audio):return slicer.slice(audio)
+
+def p2_avg_cut(audio,sr,per=3.7,overlap=0.3,tail=4):
+    i = 0
+    audios=[]
+    while (1):
+        start = int(sr * (per - overlap) * i)
+        i += 1
+        if (len(audio[start:]) > tail * sr):
+            audios.append(audio[start:start + int(per * sr)])
+        else:
+            audios.append(audio[start:])
+            break
+    return audios
+
+def p2b_get_vol(audio):return np.square(audio).mean()
+
+def p3_norm(audio,alpha=0.8,maxx=0.95):return audio / np.abs(audio).max() * (maxx * alpha) + (1-alpha) * audio
+
+def pipeline(inp_root,sr1=40000,sr2=16000,if_trim=True,if_avg_cut=True,if_norm=True,save_root1=None,save_root2=None):
+    if(save_root1==None and save_root2==None):return "No save root."
+    name2vol={}
+    infos=[]
+    names=[]
+    for name in os.listdir(inp_root):
+        try:
+            inp_path=os.path.join(inp_root,name)
+            audio=p0_load_audio(inp_path)
+        except:
+            infos.append("%s\t%s"%(name,traceback.format_exc()))
+            continue
+        if(if_trim==True):res1s=p1_trim_audio(audio)
+        else:res1s=[audio]
+        for i0,res1 in res1s:
+            if(if_avg_cut==True):res2=p2_avg_cut(res1)
+            else:res2=[res1]
+
+