def __test_gini_from_event_times(event_time_list,
expected_gini,
hop_size=0.01):
print(event_time_list)
# create scaper
sc = Scaper(10.0, FG_PATH, BG_PATH)
# add background
sc.add_background(label=('choose', []),
source_file=('choose', []),
source_time=('const', 0))
# add foreground events based on the event time list
# always use siren file since it is 26 s long, so we can choose the
# event duration flexibly
for onset, offset in event_time_list:
sc.add_event(label=('const', 'siren'),
source_file=('const', SIREN_FILE),
source_time=('const', 0),
event_time=('const', onset),
event_duration=('const', offset - onset),
snr=('uniform', 6, 30),
pitch_shift=('uniform', -3, 3),
time_stretch=None)
jam = sc._instantiate()
ann = jam.annotations[0]
gini = polyphony_gini(ann, hop_size=hop_size)
print(gini, expected_gini)
assert np.allclose([gini], [expected_gini], atol=1e-5)
def __test_gini_from_event_times(event_time_list, expected_gini,
hop_size=0.01):
print(event_time_list)
# create scaper
sc = Scaper(10.0, FG_PATH, BG_PATH)
# add background
sc.add_background(label=('choose', []),
source_file=('choose', []),
source_time=('const', 0))
# add foreground events based on the event time list
# always use siren file since it is 26 s long, so we can choose the
# event duration flexibly
for onset, offset in event_time_list:
sc.add_event(label=('const', 'siren'),
source_file=('const', SIREN_FILE),
source_time=('const', 0),
event_time=('const', onset),
event_duration=('const', offset - onset),
snr=('uniform', 6, 30),
pitch_shift=('uniform', -3, 3),
time_stretch=None)
jam = sc._instantiate()
ann = jam.annotations[0]
gini = polyphony_gini(ann, hop_size=hop_size)
print(gini, expected_gini)
assert np.allclose([gini], [expected_gini], atol=1e-5)
def test_polyphony_gini():
'''
Test computation of polyphony gini
'''
# Annotation must have namespace scaper, otherwise raise error
ann = jams.Annotation('tag_open', duration=10)
gini = pytest.raises(ScaperError, polyphony_gini, ann)
# Annotation without duration set should raise error
ann = jams.Annotation('scaper', duration=None)
gini = pytest.raises(ScaperError, polyphony_gini, ann)
# Annotation with no foreground events returns a gini of 0
sc = Scaper(10.0, FG_PATH, BG_PATH)
# add background
sc.add_background(label=('choose', []),
source_file=('choose', []),
source_time=('const', 0))
jam = sc._instantiate()
ann = jam.annotations[0]
gini = polyphony_gini(ann)
assert gini == 0
def __test_gini_from_event_times(event_time_list,
expected_gini,
hop_size=0.01):
print(event_time_list)
# create scaper
sc = Scaper(10.0, FG_PATH, BG_PATH)
# add background
sc.add_background(label=('choose', []),
source_file=('choose', []),
source_time=('const', 0))
# add foreground events based on the event time list
# always use siren file since it is 26 s long, so we can choose the
# event duration flexibly
for onset, offset in event_time_list:
sc.add_event(label=('const', 'siren'),
source_file=('const', SIREN_FILE),
source_time=('const', 0),
event_time=('const', onset),
event_duration=('const', offset - onset),
snr=('uniform', 6, 30),
pitch_shift=('uniform', -3, 3),
time_stretch=None)
jam = sc._instantiate()
ann = jam.annotations[0]
gini = polyphony_gini(ann, hop_size=hop_size)
print(gini, expected_gini)
assert np.allclose([gini], [expected_gini], atol=1e-5)
event_time_lists = ([[], [(0, 1)], [(0, 5), (5, 10)],
[(0, 10), (3, 7), (4, 6)]])
expected_ginis = [0, 0.1, 1, 0.75]
for etl, g in zip(event_time_lists, expected_ginis):
__test_gini_from_event_times(etl, g, hop_size=0.01)
for etl, g in zip(event_time_lists, expected_ginis):
__test_gini_from_event_times(etl, g, hop_size=1.0)
def test_polyphony_gini():
'''
Test computation of polyphony gini
'''
# Annotation must have namespace scaper, otherwise raise error
ann = jams.Annotation('tag_open', duration=10)
gini = pytest.raises(ScaperError, polyphony_gini, ann)
# Annotation without duration set should raise error
ann = jams.Annotation('scaper', duration=None)
gini = pytest.raises(ScaperError, polyphony_gini, ann)
# Annotation with no foreground events returns a gini of 0
sc = Scaper(10.0, FG_PATH, BG_PATH)
# add background
sc.add_background(label=('choose', []),
source_file=('choose', []),
source_time=('const', 0))
jam = sc._instantiate()
ann = jam.annotations[0]
gini = polyphony_gini(ann)
assert gini == 0
def __test_gini_from_event_times(event_time_list, expected_gini,
hop_size=0.01):
print(event_time_list)
# create scaper
sc = Scaper(10.0, FG_PATH, BG_PATH)
# add background
sc.add_background(label=('choose', []),
source_file=('choose', []),
source_time=('const', 0))
# add foreground events based on the event time list
# always use siren file since it is 26 s long, so we can choose the
# event duration flexibly
for onset, offset in event_time_list:
sc.add_event(label=('const', 'siren'),
source_file=('const', SIREN_FILE),
source_time=('const', 0),
event_time=('const', onset),
event_duration=('const', offset - onset),
snr=('uniform', 6, 30),
pitch_shift=('uniform', -3, 3),
time_stretch=None)
jam = sc._instantiate()
ann = jam.annotations[0]
gini = polyphony_gini(ann, hop_size=hop_size)
print(gini, expected_gini)
assert np.allclose([gini], [expected_gini], atol=1e-5)
event_time_lists = ([
[],
[(0, 1)],
[(0, 5), (5, 10)],
[(0, 10), (3, 7), (4, 6)]
])
expected_ginis = [0, 0.1, 1, 0.75]
for etl, g in zip(event_time_lists, expected_ginis):
__test_gini_from_event_times(etl, g, hop_size=0.01)
for etl, g in zip(event_time_lists, expected_ginis):
__test_gini_from_event_times(etl, g, hop_size=1.0)