def test_gen_uem():
ref_turns = [
Turn(0, 14, speaker_id='SPK1', file_id='FILE1'),
Turn(20, 25, speaker_id='SPK2', file_id='FILE1'),
Turn(11, 45, speaker_id='SPK3', file_id='FILE2'),
]
sys_turns = [
Turn(35, 46, speaker_id='SPK1', file_id='FILE1'),
Turn(2.5, 14, speaker_id='SPK4', file_id='FILE2'),
]
expected_uem = UEM({'FILE1': [(0, 46)], 'FILE2': [(2.5, 45)]})
assert expected_uem == gen_uem(ref_turns, sys_turns)
def main():
"""Main."""
# Parse command line arguments.
parser = ArgumentParser(
description='Score diarization from RTTM files.', add_help=True,
usage='%(prog)s [options]')
parser.add_argument(
'-r', nargs='+', default=[], metavar='STR', dest='ref_rttm_fns',
action=RefRTTMAction,
help='reference RTTM files (default: %(default)s)')
parser.add_argument(
'-R', nargs=None, metavar='STR', dest='ref_rttm_scpf',
action=RefRTTMAction,
help='reference RTTM script file (default: %(default)s)')
parser.add_argument(
'-s', nargs='+', default=[], metavar='STR', dest='sys_rttm_fns',
action=SysRTTMAction,
help='system RTTM files (default: %(default)s)')
parser.add_argument(
'-S', nargs=None, metavar='STR', dest='sys_rttm_scpf',
action=SysRTTMAction,
help='system RTTM script file (default: %(default)s)')
parser.add_argument(
'-u,--uem', nargs=None, metavar='STR', dest='uemf',
help='un-partitioned evaluation map file (default: %(default)s)')
parser.add_argument(
'--collar', nargs=None, default=0.25, type=float, metavar='FLOAT',
help='collar size in seconds for DER computaton '
'(default: %(default)s)')
parser.add_argument(
'--ignore_overlaps', action='store_true', default=False,
help='ignore overlaps when computing DER')
parser.add_argument(
'--jer_min_ref_dur', nargs=None, default=0.0, metavar='FLOAT',
help='minimum reference speaker duration for JER '
'(default: %(default)s)')
parser.add_argument(
'--step', nargs=None, default=0.010, type=float, metavar='FLOAT',
help='step size in seconds (default: %(default)s)')
parser.add_argument(
'--n_digits', nargs=None, default=2, type=int, metavar='INT',
help='number of decimal places to print (default: %(default)s)')
parser.add_argument(
'--version', action='version',
version='%(prog)s ' + VERSION)
if len(sys.argv) == 1:
parser.print_help()
sys.exit(1)
args = parser.parse_args()
# Check that at least one reference RTTM and at least one system RTTM
# was specified.
if args.ref_rttm_scpf is not None:
args.ref_rttm_fns = load_script_file(args.ref_rttm_scpf)
if args.sys_rttm_scpf is not None:
args.sys_rttm_fns = load_script_file(args.sys_rttm_scpf)
if not args.ref_rttm_fns:
error('No reference RTTMs specified.')
sys.exit(1)
if not args.sys_rttm_fns:
error('No system RTTMs specified.')
sys.exit(1)
# Load speaker/reference speaker turns and UEM. If no UEM specified,
# determine it automatically.
info('Loading speaker turns from reference RTTMs...', file=sys.stderr)
ref_turns, _ = load_rttms(args.ref_rttm_fns)
info('Loading speaker turns from system RTTMs...', file=sys.stderr)
sys_turns, _ = load_rttms(args.sys_rttm_fns)
if args.uemf is not None:
info('Loading universal evaluation map...', file=sys.stderr)
uem = load_uem(args.uemf)
else:
warn('No universal evaluation map specified. Approximating from '
'reference and speaker turn extents...')
uem = gen_uem(ref_turns, sys_turns)
# Trim turns to UEM scoring regions and merge any that overlap.
info('Trimming reference speaker turns to UEM scoring regions...',
file=sys.stderr)
ref_turns = trim_turns(ref_turns, uem)
info('Trimming system speaker turns to UEM scoring regions...',
file=sys.stderr)
sys_turns = trim_turns(sys_turns, uem)
info('Checking for overlapping reference speaker turns...',
file=sys.stderr)
ref_turns = merge_turns(ref_turns)
info('Checking for overlapping system speaker turns...',
file=sys.stderr)
sys_turns = merge_turns(sys_turns)
# Score.
info('Scoring...', file=sys.stderr)
check_for_empty_files(ref_turns, sys_turns, uem)
file_scores, global_scores = score(
ref_turns, sys_turns, uem, step=args.step,
jer_min_ref_dur=args.jer_min_ref_dur, collar=args.collar,
ignore_overlaps=args.ignore_overlaps)
print_output(
file_scores, global_scores, args.n_digits)
error('No system RTTMs specified.')
sys.exit(1)
# Load speaker/reference speaker turns and UEM. If no UEM specified,
# determine it automatically.
info('Loading speaker turns from reference RTTMs...', file=sys.stderr)
ref_turns, ref_file_ids = load_rttms(args.ref_rttm_fns)
info('Loading speaker turns from system RTTMs...', file=sys.stderr)
sys_turns, sys_file_ids = load_rttms(args.sys_rttm_fns)
if args.uemf is not None:
info('Loading universal evaluation map...', file=sys.stderr)
uem = load_uem(args.uemf)
else:
warn('No universal evaluation map specified. Approximating from '
'reference and speaker turn extents...')
uem = gen_uem(ref_turns, sys_turns)
# Trim turns to UEM scoring regions and merge any that overlap.
info('Trimming reference speaker turns to UEM scoring regions...',
file=sys.stderr)
ref_turns = trim_turns(ref_turns, uem)
info('Trimming system speaker turns to UEM scoring regions...',
file=sys.stderr)
sys_turns = trim_turns(sys_turns, uem)
info('Checking for overlapping reference speaker turns...',
file=sys.stderr)
ref_turns = merge_turns(ref_turns)
info('Checking for overlapping system speaker turns...', file=sys.stderr)
sys_turns = merge_turns(sys_turns)
# Score.