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

fix window type

This commit is contained in:
shahules786 2022-10-06 11:29:35 +05:30
parent fc7a7996bf
commit 86e56f2b50
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
enhancer/data/fileprocessor.py
View File

@ -98,6 +98,10 @@ class Fileprocessor:
return cls(clean_dir, noisy_dir, ProcessorFunctions.one_to_one) return cls(clean_dir, noisy_dir, ProcessorFunctions.one_to_one)
elif name.lower() == "dns-2020": elif name.lower() == "dns-2020":
return cls(clean_dir, noisy_dir, ProcessorFunctions.one_to_many) return cls(clean_dir, noisy_dir, ProcessorFunctions.one_to_many)
else:
raise ValueError(
f"Invalid matching function, Please use valid matching function from {MATCHING_FNS}"
)
else: else:
if matching_function not in MATCHING_FNS: if matching_function not in MATCHING_FNS:
raise ValueError( raise ValueError(

2
enhancer/inference.py
View File

@ -91,7 +91,7 @@ class Inference:
window_size: int, window_size: int,
total_frames: int, total_frames: int,
step_size: Optional[int] = None, step_size: Optional[int] = None,
window="hanning", window="hamming",
): ):
""" """
stitch batched waveform into single waveform. (Overlap-add) stitch batched waveform into single waveform. (Overlap-add)