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)