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

Merge branch 'dev' of https://github.com/shahules786/enhancer into dev-hawk

This commit is contained in:
shahules786 2022-10-19 12:38:54 +05:30
parent 7b3626c912 e4f13946e8
commit 11fbba6f77
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
enhancer/models/demucs.py
View File

@ -212,6 +212,8 @@ class Demucs(Model):
waveform = waveform.mean(dim=1, keepdim=True) waveform = waveform.mean(dim=1, keepdim=True)
std = waveform.std(dim=-1, keepdim=True) std = waveform.std(dim=-1, keepdim=True)
waveform = waveform / (self.floor + std) waveform = waveform / (self.floor + std)
else:
std = 1
length = waveform.shape[-1] length = waveform.shape[-1]
x = F.pad(waveform, (0, self.get_padding_length(length) - length)) x = F.pad(waveform, (0, self.get_padding_length(length) - length))
if self.hparams.resample > 1: if self.hparams.resample > 1:
@ -244,7 +246,7 @@ class Demucs(Model):
) )
out = x[..., :length] out = x[..., :length]
return out return std * out
def get_padding_length(self, input_length): def get_padding_length(self, input_length):