| | import os, fnmatch |
| | import numpy as np |
| | import random |
| | import soundfile as sf |
| | from scipy.io.wavfile import write |
| | |
| | import RT60 |
| |
|
| | folder_path = "/cephfs/anton/room-impulse-responses/AIR/RWCP_REVERB_AACHEN/real_rirs_isotropic_noises/" |
| | final_path = "/cephfs/anton/room-impulse-responses/AIR/RWCP_REVERB_AACHEN/AACHEN/" |
| | tfs =16000 |
| | file_label = open("RT60.txt","w") |
| |
|
| | for root, dirnames, filenames in os.walk(folder_path): |
| | for filename in filenames: |
| | if filename.endswith(".wav"): |
| | ACE_Path = os.path.join(root, filename) |
| | wave,fs = sf.read(ACE_Path) |
| | channel = int(wave.size/len(wave)) |
| |
|
| | if(channel == 1): |
| | wave_single = wave |
| | max_loc = np.where(wave_single == np.amax(wave_single)) |
| | min_loc = np.where(wave_single == np.amin(wave_single)) |
| | start = min(max_loc[0][0],min_loc[0][0]) |
| | wave_single =wave_single[start:len(wave_single)] |
| | T60_val = RT60.t60_impulse(wave_single,tfs) |
| | |
| | if(T60_val<1): |
| | file_label.write(str(T60_val)+"\n") |
| | save_path = final_path+ filename |
| | write(save_path,tfs,wave_single.astype(np.float32)) |
| | else: |
| | for n in range(channel): |
| | wave_single = wave[:,n] |
| | max_loc = np.where(wave_single == np.amax(wave_single)) |
| | min_loc = np.where(wave_single == np.amin(wave_single)) |
| | start = min(max_loc[0][0],min_loc[0][0]) |
| | wave_single =wave_single[start:len(wave_single)] |
| | T60_val = RT60.t60_impulse(wave_single,tfs) |
| | |
| | if(T60_val<1): |
| | file_label.write(str(T60_val)+"\n") |
| | save_path = final_path+filename+str(n)+".wav" |
| | write(save_path,tfs,wave_single.astype(np.float32)) |
| | |
