64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
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]
|
|
|
|
|