def test_add_rir_noise_aecres(self): wav_path = str(Path(PACKAGE_OPS_DIR).joinpath('data/sm1_cln.wav')) # reset path of noise && rir data_path = str(Path(PACKAGE_OPS_DIR).joinpath('data')) + '/' noise_file = data_path + 'noiselist_new.scp' change_file_path(data_path, 'noiselist.scp', 'noiselist_new.scp') rir_file = data_path + 'rirlist_new.scp' change_file_path(data_path, 'rirlist.scp', 'rirlist_new.scp') with self.cached_session(use_gpu=False, force_gpu=False) as sess: read_wav = ReadWav.params().instantiate() input_data, sample_rate = read_wav(wav_path) config = { 'if_add_noise': True, 'noise_filelist': noise_file, 'if_add_rir': True, 'rir_filelist': rir_file } add_rir_noise_aecres = Add_rir_noise_aecres.params(config).instantiate() add_rir_noise_aecres_test = add_rir_noise_aecres(input_data, sample_rate) print('Clean Data:', input_data.eval()) print('Noisy Data:', add_rir_noise_aecres_test.eval()) new_noise_file = data_path + 'sm1_cln_noisy.wav' write_wav = WriteWav.params().instantiate() writewav_op = write_wav(new_noise_file, add_rir_noise_aecres_test / 32768, sample_rate) sess.run(writewav_op)
def test_write_wav(self): wav_path = str( Path(os.environ['MAIN_ROOT']).joinpath('delta/layers/ops/data/sm1_cln.wav')) with self.session() as sess: read_wav = ReadWav.params().instantiate() input_data, sample_rate = read_wav(wav_path) write_wav = WriteWav.params().instantiate() new_path = str(Path(os.environ['MAIN_ROOT']).joinpath('delta/layers/ops/data/sm1_cln_new.wav')) writewav_op = write_wav(new_path, input_data, sample_rate) sess.run(writewav_op) test_data, test_sample_rate = read_wav(new_path) self.assertAllEqual(input_data.eval(), test_data.eval()) self.assertAllEqual(sample_rate.eval(),test_sample_rate.eval())
def test_write_wav(self): wav_path = str(Path(PACKAGE_OPS_DIR).joinpath('data/sm1_cln.wav')) with self.cached_session(use_gpu=False, force_gpu=False) as sess: read_wav = ReadWav.params({'speed': 1.0}).instantiate() input_data, sample_rate = read_wav(wav_path) input_data = input_data write_wav = WriteWav.params().instantiate() new_path = str( Path(PACKAGE_OPS_DIR).joinpath('data/sm1_cln_speed.wav')) writewav_op = write_wav(new_path, input_data, sample_rate) sess.run(writewav_op) test_data, test_sample_rate = read_wav(new_path) self.assertAllEqual(input_data.eval(), test_data.eval()) self.assertAllEqual(sample_rate.eval(), test_sample_rate.eval())
def test_write_wav(self): wav_path = str( Path(PACKAGE_ROOT_DIR).joinpath('layers/ops/data/sm1_cln.wav')) with self.cached_session(use_gpu=False, force_gpu=False) as sess: read_wav = ReadWav.params().instantiate() input_data, sample_rate = read_wav(wav_path) input_data = input_data / 32768 write_wav = WriteWav.params().instantiate() new_path = str( Path(PACKAGE_ROOT_DIR).joinpath( 'layers/ops/data/sm1_cln_new.wav')) writewav_op = write_wav(new_path, input_data, sample_rate) sess.run(writewav_op) test_data, test_sample_rate = read_wav(new_path) test_data = test_data / 32768 self.assertAllEqual(input_data.eval(), test_data.eval()) self.assertAllEqual(sample_rate.eval(), test_sample_rate.eval())
def __init__(self, config: dict): super().__init__(config) self.add_noise = Add_rir_noise_aecres(config) self.read_wav = ReadWav(config) self.write_wav = WriteWav(config)