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

add tqdm

Browse Source
This commit is contained in:
Stardust·减
2023-09-05 09:57:08 +08:00
committed by GitHub
parent bb4ebb80f7
commit 316e0ba345
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
data_utils.py
View File

@@ -4,6 +4,8 @@ import random
import numpy as np import numpy as np
import torch import torch
import torch.utils.data import torch.utils.data
from tqdm import tqdm
import commons import commons
from mel_processing import spectrogram_torch, mel_spectrogram_torch, spec_to_mel_torch from mel_processing import spectrogram_torch, mel_spectrogram_torch, spec_to_mel_torch
from utils import load_wav_to_torch, load_filepaths_and_text from utils import load_wav_to_torch, load_filepaths_and_text
@@ -55,7 +57,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
audiopaths_sid_text_new = [] audiopaths_sid_text_new = []
lengths = [] lengths = []
skipped = 0 skipped = 0
for _id, spk, language, text, phones, tone, word2ph in self.audiopaths_sid_text: for _id, spk, language, text, phones, tone, word2ph in tqdm(self.audiopaths_sid_text):
audiopath = f'{_id}' audiopath = f'{_id}'
if self.min_text_len <= len(phones) and len(phones) <= self.max_text_len: if self.min_text_len <= len(phones) and len(phones) <= self.max_text_len:
phones = phones.split(" ") phones = phones.split(" ")