Delete denoise_audio.py
This commit is contained in:
@@ -1,31 +0,0 @@
|
||||
import os
|
||||
from concurrent.futures import ProcessPoolExecutor
|
||||
|
||||
import torchaudio
|
||||
|
||||
WORKERS = 3 # 6G显存+16G内存,请根据机器配置调整
|
||||
raw_audio_dir = "./raw_audio/"
|
||||
denoise_audio_dir = "./denoised_audio/"
|
||||
filelist = list(os.walk(raw_audio_dir))[0][2]
|
||||
|
||||
|
||||
def denoise_audio(file):
|
||||
if file.endswith(".wav"):
|
||||
os.system(f"demucs --two-stems=vocals {raw_audio_dir}{file}")
|
||||
|
||||
file = file.replace(".wav", "")
|
||||
wav, sr = torchaudio.load(f"./separated/htdemucs/{file}/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(denoise_audio_dir + file + ".wav", wav, 22050, channels_first=True)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
with ProcessPoolExecutor(max_workers=WORKERS) as executor:
|
||||
executor.map(denoise_audio, filelist)
|
||||
Reference in New Issue
Block a user