This commit is contained in:
speech_asr 2023-03-15 16:44:59 +08:00
parent 7c6ed3830a
commit dd4946a50d

View File

@ -237,7 +237,8 @@ def inference_modelscope(
results = speech2diar(**batch)
# post process
a = medfilt(results[0], (11, 1))
a = results[0].cpu().numpy()
a = medfilt(a, (11, 1))
rst = []
for spkid, frames in enumerate(a.T):
frames = np.pad(frames, (1, 1), 'constant')