def setUpClass(cls): env_config_path = os.path.abspath(os.path.join( os.path.dirname(__file__), '..', '..', '..', '.env')) load_dotenv(env_config_path) # https://freewavesamples.com/yamaha-v50-rock-beat-120-bpm cls.sample_wav_file = os.environ.get("DATA_DIR") + 'Yamaha-V50-Rock-Beat-120bpm.wav' # https://en.wikipedia.org/wiki/Colors_of_noise cls.noise_wav_file = os.environ.get("DATA_DIR") + 'Pink_noise.ogg' cls.audio, cls.sampling_rate = AudioLoader.load_audio(cls.sample_wav_file) cls.noise, cls.noise_sampling_rate = AudioLoader.load_audio(cls.noise_wav_file)
def setUpClass(cls): env_config_path = os.path.abspath(os.path.join( os.path.dirname(__file__), '..', '..', '..', '.env')) load_dotenv(env_config_path) # https://freewavesamples.com/yamaha-v50-rock-beat-120-bpm cls.sample_wav_file = os.environ.get("DATA_DIR") + 'Yamaha-V50-Rock-Beat-120bpm.wav' cls.audio, cls.sampling_rate = AudioLoader.load_audio(cls.sample_wav_file)
def test_substitute(self): audio, sampling_rate = AudioLoader.load_audio(self.sample_wav_file) aug = naa.ShiftAug(sampling_rate, duration=0.5) augmented_audio = aug.augment(audio) self.assertFalse(np.array_equal(audio, augmented_audio)) self.assertTrue(len(audio), len(augmented_audio))
def test_multi_thread(self): audio, sampling_rate = AudioLoader.load_audio(self.sample_wav_file) n = 3 augs = [ naa.CropAug(sampling_rate=sampling_rate), naa.PitchAug(sampling_rate=sampling_rate) ] for num_thread in [1, 3]: for aug in augs: augmented_data = aug.augment(audio, n=n, num_thread=num_thread) self.assertEqual(len(augmented_data), n)
def test_audio(self): audio, sampling_rate = AudioLoader.load_audio(self.sample_wav_file) flow = naf.Sequential([ naa.NoiseAug(), naa.PitchAug(sampling_rate=sampling_rate, factor=(0.2, 1.5)), naa.ShiftAug(sampling_rate=sampling_rate, duration=2), naa.SpeedAug(factor=(1.5, 3)) ]) augmented_audio = flow.augment(audio) self.assertFalse(np.array_equal(audio, augmented_audio)) self.assertTrue(len(audio), len(augmented_audio))
def test_audio(self): # https://freewavesamples.com/yamaha-v50-rock-beat-120-bpm sample_wav_file = os.path.abspath( os.path.join(os.path.dirname(__file__), '..', '..', 'data', 'Yamaha-V50-Rock-Beat-120bpm.wav')) audio, sampling_rate = AudioLoader.load_audio(sample_wav_file) flow = naf.Sequential([ naa.NoiseAug(), naa.PitchAug(sampling_rate=sampling_rate, factor=(0.2, 1.5)), naa.ShiftAug(sampling_rate=sampling_rate, duration=2), naa.SpeedAug(factor=(1.5, 3)) ]) augmented_audio = flow.augment(audio) self.assertFalse(np.array_equal(audio, augmented_audio)) self.assertTrue(len(audio), len(augmented_audio))