jshiffer/Retrieval-based-Voice-Conversion
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

format

Browse Source
This commit is contained in:
Ftps
2024-01-21 01:15:44 +09:00
parent c839cab165
commit 7e0326146d
7 changed files with 17 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
rvc/lib/rmvpe.py
View File

@@ -18,15 +18,13 @@ try:
except Exception: # pylint: disable=broad-exception-caught
pass
import logging
from time import time as ttime
import torch.nn as nn
import torch.nn.functional as F
from librosa.filters import mel
from librosa.util import normalize, pad_center, tiny
from scipy.signal import get_window
from librosa.filters import mel
from time import time as ttime
logger = logging.getLogger(__name__)
@@ -161,7 +159,6 @@ class STFT(torch.nn.Module):
return reconstruction
class BiGRU(nn.Module):
def __init__(self, input_features, hidden_features, num_layers):
super(BiGRU, self).__init__()
@@ -415,8 +412,6 @@ class E2E(nn.Module):
return x
class MelSpectrogram(torch.nn.Module):
def __init__(
self,

18
rvc/lib/slicer2.py
View File

@@ -64,11 +64,11 @@ class Slicer:
def _apply_slice(self, waveform, begin, end):
if len(waveform.shape) > 1:
return waveform[
:, begin * self.hop_size: min(waveform.shape[1], end * self.hop_size)
:, begin * self.hop_size : min(waveform.shape[1], end * self.hop_size)
]
else:
return waveform[
begin * self.hop_size: min(waveform.shape[0], end * self.hop_size)
begin * self.hop_size : min(waveform.shape[0], end * self.hop_size)
]
# @timeit
@@ -106,7 +106,7 @@ class Slicer:
continue
# Need slicing. Record the range of silent frames to be removed.
if i - silence_start <= self.max_sil_kept:
pos = rms_list[silence_start: i + 1].argmin() + silence_start
pos = rms_list[silence_start : i + 1].argmin() + silence_start
if silence_start == 0:
sil_tags.append((0, pos))
else:
@@ -114,17 +114,17 @@ class Slicer:
clip_start = pos
elif i - silence_start <= self.max_sil_kept * 2:
pos = rms_list[
i - self.max_sil_kept: silence_start + self.max_sil_kept + 1
i - self.max_sil_kept : silence_start + self.max_sil_kept + 1
].argmin()
pos += i - self.max_sil_kept
pos_l = (
rms_list[
silence_start: silence_start + self.max_sil_kept + 1
silence_start : silence_start + self.max_sil_kept + 1
].argmin()
+ silence_start
)
pos_r = (
rms_list[i - self.max_sil_kept: i + 1].argmin()
rms_list[i - self.max_sil_kept : i + 1].argmin()
+ i
- self.max_sil_kept
)
@@ -137,12 +137,12 @@ class Slicer:
else:
pos_l = (
rms_list[
silence_start: silence_start + self.max_sil_kept + 1
silence_start : silence_start + self.max_sil_kept + 1
].argmin()
+ silence_start
)
pos_r = (
rms_list[i - self.max_sil_kept: i + 1].argmin()
rms_list[i - self.max_sil_kept : i + 1].argmin()
+ i
- self.max_sil_kept
)
@@ -159,7 +159,7 @@ class Slicer:
and total_frames - silence_start >= self.min_interval
):
silence_end = min(total_frames, silence_start + self.max_sil_kept)
pos = rms_list[silence_start: silence_end + 1].argmin() + silence_start
pos = rms_list[silence_start : silence_end + 1].argmin() + silence_start
sil_tags.append((pos, total_frames + 1))
# Apply and return slices.
if len(sil_tags) == 0: