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

make resample static

This commit is contained in:
shahules786 2022-09-05 17:12:20 +05:30
parent 409afc31fc
commit 8a43354cb0
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
enhancer/utils/io.py
View File

@ -42,7 +42,7 @@ class Audio:
if self.mono: if self.mono:
audio = self.convert_mono(audio) audio = self.convert_mono(audio)
resampled_audio = self.resample_audio(audio,sampling_rate) resampled_audio = self.__class__.resample_audio(audio,self.sampling_rate,sampling_rate)
if self.return_tensor: if self.return_tensor:
return torch.tensor(resampled_audio) return torch.tensor(resampled_audio)
else: else:
@ -59,12 +59,13 @@ class Audio:
return audio return audio
@staticmethod
def resample_audio( def resample_audio(
self,
audio, audio,
sampling_rate sr:int,
target_sr:int
): ):
if self.sampling_rate!=sampling_rate: if sr!=target_sr:
audio = librosa.resample(audio,orig_sr=sampling_rate,target_sr=self.sampling_rate) audio = librosa.resample(audio,orig_sr=sr,target_sr=target_sr)
return audio return audio