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

add logger

Browse Source
This commit is contained in:
Stardust·减
2023-09-05 19:50:24 +08:00
committed by GitHub
parent 89b6c5d589
commit ff0c76dccb
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
data_utils.py
View File

@@ -5,7 +5,7 @@ import numpy as np
import torch import torch
import torch.utils.data import torch.utils.data
from tqdm import tqdm from tqdm import tqdm
from loguru import logger
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
@@ -57,6 +57,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
audiopaths_sid_text_new = [] audiopaths_sid_text_new = []
lengths = [] lengths = []
skipped = 0 skipped = 0
logger.info("Init dataset...")
for _id, spk, language, text, phones, tone, word2ph in tqdm(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:
@@ -67,7 +68,7 @@ class TextAudioSpeakerLoader(torch.utils.data.Dataset):
lengths.append(os.path.getsize(audiopath) // (2 * self.hop_length)) lengths.append(os.path.getsize(audiopath) // (2 * self.hop_length))
else: else:
skipped += 1 skipped += 1
print("skipped: ", skipped, ", total: ", len(self.audiopaths_sid_text)) logger.info("skipped: " + skipped + ", total: " + len(self.audiopaths_sid_text))
self.audiopaths_sid_text = audiopaths_sid_text_new self.audiopaths_sid_text = audiopaths_sid_text_new
self.lengths = lengths self.lengths = lengths