From fbe46caa3dc363378106a6b7f9f170164bcd0351 Mon Sep 17 00:00:00 2001
From: Plachtaa <112609742+Plachtaa@users.noreply.github.com>
Date: Sun, 5 Mar 2023 16:33:29 +0800
Subject: [PATCH] Add files via upload

---
 denoise_audio.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)
 create mode 100644 denoise_audio.py

diff --git a/denoise_audio.py b/denoise_audio.py
new file mode 100644
index 0000000..a25f076
--- /dev/null
+++ b/denoise_audio.py
@@ -0,0 +1,18 @@
+import os
+import torchaudio
+raw_audio_dir = "./raw_audio/"
+denoise_audio_dir = "./denoised_audio/"
+filelist = list(os.walk(raw_audio_dir))[0][2]
+
+for file in filelist:
+    if file.endswith(".wav"):
+        os.system(f"demucs --two-stems=vocals {raw_audio_dir}{file}")
+for file in filelist:
+    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)
\ No newline at end of file