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