From 7d5b2f8547e76c6cf467c546ff15ff51cec5cd07 Mon Sep 17 00:00:00 2001 From: Plachta Date: Sun, 26 Feb 2023 20:31:00 +0800 Subject: [PATCH] updated pipeline --- demucs_denoise.py | 31 ------------------------------- 1 file changed, 31 deletions(-) delete mode 100644 demucs_denoise.py diff --git a/demucs_denoise.py b/demucs_denoise.py deleted file mode 100644 index 718ca71..0000000 --- a/demucs_denoise.py +++ /dev/null @@ -1,31 +0,0 @@ -import os -import torch -import torchaudio -import argparse -if __name__ == "__main__": - parser = argparse.ArgumentParser() - parser.add_argument('--denoise_user', type=bool, default=False, - help='whether to denoise user recorded voice') - parser.add_argument('--denoise_character', type=bool, default=False, - help='whether to denoise uploaded character voice') - args = parser.parse_args() - if args.denoise_user: - audio_dir = "./user_voice/" - wavfiles = [] - for filename in list(os.walk(audio_dir))[0][2]: - if filename.endswith(".wav"): - wavfiles.append(filename) - - # denoise with demucs - for i, wavfile in enumerate(wavfiles): - os.system(f"demucs --two-stems=vocals {audio_dir}{wavfile}") - - # read & store the denoised vocals back - for wavfile in wavfiles: - i = wavfile.strip(".wav") - wav, sr = torchaudio.load(f"./separated/htdemucs/{i}/vocals.wav", frame_offset=0, num_frames=-1, normalize=True, channels_first=True) - # merge two channels into one - wav = wav.mean(dim=0).unsqueeze(0) - if sr != 22050: - wav = torchaudio.transforms.Resample(orig_freq=sr, new_freq=22050)(wav) - torchaudio.save(f"./user_voice/{i}.wav", wav, 22050, channels_first=True) \ No newline at end of file