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

Update resample.py

Browse Source
This commit is contained in:
Stardust·减
2023-12-14 16:26:41 +08:00
committed by GitHub
parent 7a2053ea9f
commit bf9ab35e6e
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
resample.py
View File

@@ -10,11 +10,11 @@ from config import config
def process(item): def process(item):
wav_name, args = item spkdir, wav_name, args = item
wav_path = os.path.join(args.in_dir, wav_name) wav_path = os.path.join(args.in_dir, spkdir, wav_name)
if os.path.exists(wav_path) and wav_path.lower().endswith(".wav"): if os.path.exists(wav_path) and wav_path.lower().endswith(".wav"):
wav, sr = librosa.load(wav_path, sr=args.sr) wav, sr = librosa.load(wav_path, sr=args.sr)
soundfile.write(os.path.join(args.out_dir, wav_name), wav, sr) soundfile.write(os.path.join(args.out_dir, spkdir, wav_name), wav, sr)
if __name__ == "__main__": if __name__ == "__main__":
@@ -54,11 +54,15 @@ if __name__ == "__main__":
tasks = [] tasks = []
for dirpath, _, filenames in os.walk(args.in_dir): for dirpath, _, filenames in os.walk(args.in_dir):
if not os.path.isdir(args.out_dir): # 子级目录
os.makedirs(args.out_dir, exist_ok=True) spk_dir = os.path.relpath(dirpath, args.in_dir)
spk_dir_out = os.path.join(args.out_dir, spk_dir)
if not os.path.isdir(spk_dir_out):
os.makedirs(spk_dir_out, exist_ok=True)
for filename in filenames: for filename in filenames:
if filename.lower().endswith(".wav"): if filename.lower().endswith(".wav"):
tasks.append((filename, args)) twople = (spk_dir, filename, args)
tasks.append(twople)
for _ in tqdm( for _ in tqdm(
pool.imap_unordered(process, tasks), pool.imap_unordered(process, tasks),