| | 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): |
| | 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] |
| |
|
| |
|
| |
|
