tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Update data_utils.py

Browse Source
This commit is contained in:
Stardust·减
2023-08-21 08:39:05 +08:00
committed by GitHub
parent f82ba45ab8
commit 3dab1b60b1
1 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
data_utils.py
View File

@@ -94,15 +94,12 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
spec = torch.load(spec_filename) spec = torch.load(spec_filename)
else: else:
if self.use_mel_spec_posterior: if self.use_mel_spec_posterior:
''' TODO : (need verification) # if os.path.exists(filename.replace(".wav", ".spec.pt")):
if linear spec exists convert to # # spec, n_fft, num_mels, sampling_rate, fmin, fmax
mel from existing linear spec (uncomment below lines) ''' # spec = spec_to_mel_torch(
if os.path.exists(filename.replace(".wav", ".spec.pt")): # torch.load(filename.replace(".wav", ".spec.pt")),
#spec, n_fft, num_mels, sampling_rate, fmin, fmax # self.filter_length, self.n_mel_channels, self.sampling_rate,
#spec = spec_to_mel_torch( # self.hparams.mel_fmin, self.hparams.mel_fmax)
#torch.load(filename.replace(".wav", ".spec.pt")),
#self.filter_length, self.n_mel_channels, self.sampling_rate,
#self.hparams.mel_fmin, self.hparams.mel_fmax)
spec = mel_spectrogram_torch(audio_norm, self.filter_length, spec = mel_spectrogram_torch(audio_norm, self.filter_length,
self.n_mel_channels, self.sampling_rate, self.hop_length, self.n_mel_channels, self.sampling_rate, self.hop_length,
self.win_length, self.hparams.mel_fmin, self.hparams.mel_fmax, center=False) self.win_length, self.hparams.mel_fmin, self.hparams.mel_fmax, center=False)