Commit 31911a3e authored by Anthony Larcher's avatar Anthony Larcher
Browse files

cleaning

parent 0c053d7f
......@@ -507,14 +507,14 @@ class IdMapSetPerSpeaker(Dataset):
self.transform_number = transform_number
self.noise_df = None
if "add_noise" in self.transform:
if "add_noise" in self.transformation:
# Load the noise dataset, filter according to the duration
noise_df = pandas.read_csv(self.transformation["add_noise"]["noise_db_csv"])
tmp_df = noise_df.loc[noise_df['duration'] > self.duration]
self.noise_df = tmp_df['file_id'].tolist()
self.rir_df = None
if "add_reverb" in self.transform:
if "add_reverb" in self.transformation:
# load the RIR database
tmp_rir_df = pandas.read_csv(self.transformation["add_reverb"]["rir_db_csv"])
self.rir_df = zip(tmp_rir_df['file_id'].tolist(), tmp_rir_df['channel'].tolist())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment