Merge pull request #336 from Miuzarte/4.1-Stable

why not rich.progress
This commit is contained in:
謬紗特 2023-07-23 09:16:43 +08:00 committed by GitHub
commit b0e7e83880
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions

View File

@ -7,7 +7,7 @@ from multiprocessing import cpu_count
import librosa import librosa
import numpy as np import numpy as np
from scipy.io import wavfile from scipy.io import wavfile
from tqdm import tqdm from rich.progress import track
def load_wav(wav_path): def load_wav(wav_path):
@ -81,7 +81,7 @@ def process_all_speakers():
if os.path.isdir(spk_dir): if os.path.isdir(spk_dir):
print(spk_dir) print(spk_dir)
futures = [executor.submit(process, (spk_dir, i, args)) for i in os.listdir(spk_dir) if i.endswith("wav")] futures = [executor.submit(process, (spk_dir, i, args)) for i in os.listdir(spk_dir) if i.endswith("wav")]
for _ in tqdm(concurrent.futures.as_completed(futures), total=len(futures)): for _ in track(concurrent.futures.as_completed(futures), total=len(futures), description="resampling:"):
pass pass