WindyMadman/so-vits-svc
1
0
Fork 0
mirror of https://github.com/svc-develop-team/so-vits-svc.git synced 2025-01-09 12:37:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

Parallelized loudness calculation

Browse Source
This commit is contained in:
ylzz1997 2023-06-17 01:18:38 +08:00
parent 2de65ca297
commit 1805061e56
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils.py
View File

@ -534,6 +534,6 @@ class Volume_Extractor:
n_frames = int(audio.size(-1) // self.hop_size)
audio2 = audio ** 2
audio2 = torch.nn.functional.pad(audio2, (int(self.hop_size // 2), int((self.hop_size + 1) // 2)), mode = 'reflect')
volume = torch.FloatTensor([torch.mean(audio2[:,int(n * self.hop_size) : int((n + 1) * self.hop_size)]) for n in range(n_frames)])
volume = torch.nn.functional.unfold(audio2[:,None,None,:],(1,self.hop_size),stride=self.hop_size)[:,:,:n_frames].mean(dim=1)[0]
volume = torch.sqrt(volume)
return volume