def test_augmentation_chain_randomized(
    recording: Recording,
    rir: Recording,
    target_sampling_rate: int,
    sp_factor: float,
    vp_factor: float,
    reverb: bool,
    resample_first: bool,
    cut_duration: Seconds,
):
    if resample_first:
        recording_aug = (recording.resample(target_sampling_rate).
                         perturb_speed(sp_factor).perturb_volume(vp_factor))
    else:
        recording_aug = (recording.perturb_speed(sp_factor).resample(
            target_sampling_rate).perturb_volume(vp_factor))
    if reverb:
        recording_aug = recording_aug.reverb_rir(rir)

    audio_aug = recording_aug.load_audio()
    assert audio_aug.shape[1] == recording_aug.num_samples

    cut_aug = MonoCut(
        id="dummy",
        start=0.5125,
        duration=cut_duration,
        channel=0,
        recording=recording_aug,
    )
    assert cut_aug.load_audio().shape[1] == cut_aug.num_samples
Exemple #2
0
def test_augmentation_chain_randomized(
        target_sampling_rate: int,
        sp_factor: float,
        resample_first: bool,
        cut_duration: Seconds
):
    recording = Recording.from_file('test/fixtures/libri/libri-1088-134315-0000.wav')

    if resample_first:
        recording_aug = recording.resample(target_sampling_rate).perturb_speed(sp_factor)
    else:
        recording_aug = recording.perturb_speed(sp_factor).resample(target_sampling_rate)

    audio_aug = recording_aug.load_audio()
    assert audio_aug.shape[1] == recording_aug.num_samples

    cut_aug = MonoCut(id='dummy', start=0.5125, duration=cut_duration, channel=0, recording=recording_aug)
    assert cut_aug.load_audio().shape[1] == cut_aug.num_samples