Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fileprocessor

This commit is contained in:
shahules786 2022-09-03 10:49:43 +05:30
parent afdf42f05a
commit 7a112786b4
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
enhancer/utils/fileprocessor.py
View File

@ -43,7 +43,7 @@ class Fileprocessor:
@staticmethod @staticmethod
def match_vtck(clean_path,noisy_path,sr): def match_vtck(clean_path,noisy_path,sr):
matching_wavfiles = dict() matching_wavfiles = list()
clean_filenames = [file.split('/')[-1] for file in glob.glob(os.path.join(clean_path,"*.wav"))] clean_filenames = [file.split('/')[-1] for file in glob.glob(os.path.join(clean_path,"*.wav"))]
noisy_filenames = [file.split('/')[-1] for file in glob.glob(os.path.join(noisy_path,"*.wav"))] noisy_filenames = [file.split('/')[-1] for file in glob.glob(os.path.join(noisy_path,"*.wav"))]
common_filenames = np.intersect1d(noisy_filenames,clean_filenames) common_filenames = np.intersect1d(noisy_filenames,clean_filenames)
@ -55,10 +55,9 @@ class Fileprocessor:
if ((clean_file.shape[-1]==noisy_file.shape[-1]) and if ((clean_file.shape[-1]==noisy_file.shape[-1]) and
(sr_clean==sr) and (sr_clean==sr) and
(sr_noisy==sr)): (sr_noisy==sr)):
matching_wavfiles.update( matching_wavfiles.append(
{os.path.join(clean_path,file_name):{"noisy":os.path.join(noisy_path,file_name), {"clean":os.path.join(clean_path,file_name),"noisy":os.path.join(noisy_path,file_name),
"duration":clean_file.shape[-1]/sr} "duration":clean_file.shape[-1]/sr}
}
) )
return matching_wavfiles return matching_wavfiles
@ -77,9 +76,8 @@ class Fileprocessor:
(sr_clean==sr) and (sr_clean==sr) and
(sr_noisy==sr)): (sr_noisy==sr)):
matching_wavfiles.update( matching_wavfiles.update(
{os.path.join(clean_path,clean_file):{"noisy":noisy_file, {"clean":os.path.join(clean_path,clean_file),"noisy":noisy_file,
"duration":clean_file.shape[-1]/sr} "duration":clean_file.shape[-1]/sr}
}
) )
return matching_wavfiles return matching_wavfiles