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