chrissewell/whisperX
1
0
Fork 0
mirror of https://github.com/m-bain/whisperX.git synced 2025-07-01 18:17:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

v2 lets goo

Browse Source
This commit is contained in:
Max Bain
2023-04-01 00:10:45 +01:00
parent bc2776017e
commit 189aeac83e
2 changed files with 2 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
whisperx/vad.py
View File

@ -301,18 +301,4 @@ def merge_chunks(segments, chunk_size):
"end": curr_end,
"segments": seg_idxs,
})
return merged_segments
if __name__ == "__main__":
import pandas as pd
input_fp = "tt298650_sync.wav"
df = pd.read_csv(f"/work/maxbain/tmp/{input_fp}.sad", sep=" ", header=None)
print(len(df))
N = 0.15
g = df[0].sub(df[1].shift())
input_base = input_fp.split('.')[0]
df = df.groupby(g.gt(N).cumsum()).agg({0:'min', 1:'max'})
df.to_csv(f"/work/maxbain/tmp/{input_base}.lab", header=None, index=False, sep=" ")
print(df)
import pdb; pdb.set_trace()
return merged_segments