File size: 767 Bytes
d2f59ca | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 | """LS-EEND streaming speaker diarization on Axera AX650N.
Typical use:
from ls_eend_sdk import diarize
result = diarize('meeting.wav', 'models/streaming_step.axmodel')
print(result['rttm'], result['speakers'])
"""
from .diarize import diarize
from .feature import extract_features, load_audio, wav_to_features, FRAME_SEC
from .postprocess import to_activity, to_segments, write_rttm
from .session import CONV_DELAY, INPUT_NAMES, OUTPUT_NAMES, StreamingDiarizer
__version__ = '1.0.0'
__all__ = [
'diarize',
'StreamingDiarizer',
'load_audio',
'extract_features',
'wav_to_features',
'to_activity',
'to_segments',
'write_rttm',
'CONV_DELAY',
'FRAME_SEC',
'INPUT_NAMES',
'OUTPUT_NAMES',
'__version__',
]
|