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

优化重采样

Browse Source
This commit is contained in:
Stardust·减
2023-08-29 14:47:13 +08:00
committed by GitHub
parent f140b7dafe
commit 253f1479f5
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
resample.py
View File

@@ -11,7 +11,6 @@ from tqdm import tqdm
def process(item): def process(item):
spkdir, wav_name, args = item spkdir, wav_name, args = item
# speaker 's5', 'p280', 'p315' are excluded,
speaker = spkdir.replace("\\", "/").split("/")[-1] speaker = spkdir.replace("\\", "/").split("/")[-1]
wav_path = os.path.join(args.in_dir, speaker, wav_name) wav_path = os.path.join(args.in_dir, speaker, wav_name)
if os.path.exists(wav_path) and '.wav' in wav_path: if os.path.exists(wav_path) and '.wav' in wav_path:
@@ -27,11 +26,12 @@ def process(item):
if __name__ == "__main__": if __name__ == "__main__":
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--sr2", type=int, default=22050, help="sampling rate") parser.add_argument("--sr", type=int, default=44100, help="sampling rate")
parser.add_argument("--in_dir", type=str, default="./raw", help="path to source dir") parser.add_argument("--in_dir", type=str, default="./raw", help="path to source dir")
parser.add_argument("--out_dir2", type=str, default="./dataset", help="path to target dir") parser.add_argument("--out_dir", type=str, default="./dataset", help="path to target dir")
args = parser.parse_args() args = parser.parse_args()
processs = 8 # processs = 8
processs = cpu_count()-2 if cpu_count() >4 else 1
pool = Pool(processes=processs) pool = Pool(processes=processs)
for speaker in os.listdir(args.in_dir): for speaker in os.listdir(args.in_dir):