heimoshuiyu/ultimatevocalremovergui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add files via upload

Browse Source
This commit is contained in:
Anjok07
2023-04-12 02:13:30 -05:00
committed by GitHub
parent 18d32660db
commit 6ffd7a244e
2 changed files with 21 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
lib_v5/vr_network/nets.py
View File

@@ -118,7 +118,7 @@ class CascadedASPPNet(nn.Module):
self.offset = 128
def forward(self, x, aggressiveness=None):
def forward(self, x):
mix = x.detach()
x = x.clone()
@@ -155,17 +155,12 @@ class CascadedASPPNet(nn.Module):
mode='replicate')
return mask * mix, aux1 * mix, aux2 * mix
else:
if aggressiveness:
mask[:, :, :aggressiveness['split_bin']] = torch.pow(mask[:, :, :aggressiveness['split_bin']], 1 + aggressiveness['value'] / 3)
mask[:, :, aggressiveness['split_bin']:] = torch.pow(mask[:, :, aggressiveness['split_bin']:], 1 + aggressiveness['value'])
return mask# * mix
return mask * mix
def predict(self, x_mag, aggressiveness=None):
h = self.forward(x_mag, aggressiveness)
def predict_mask(self, x):
mask = self.forward(x)
if self.offset > 0:
h = h[:, :, :, self.offset:-self.offset]
assert h.size()[3] > 0
mask = mask[:, :, :, self.offset:-self.offset]
return h
return mask