File size: 1,037 Bytes
32b2aaa |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
import argparse
from pathlib import Path
import torch
import torchaudio
from .inference import denoise
@torch.inference_mode()
def main():
parser = argparse.ArgumentParser()
parser.add_argument("in_dir", type=Path, help="Path to input audio folder")
parser.add_argument("out_dir", type=Path, help="Output folder")
parser.add_argument("--run_dir", type=Path, default="runs/denoiser", help="Path to run folder")
parser.add_argument("--suffix", type=str, default=".wav", help="File suffix")
parser.add_argument("--device", type=str, default="cuda", help="Device")
args = parser.parse_args()
for path in args.in_dir.glob(f"**/*{args.suffix}"):
print(f"Processing {path} ..")
dwav, sr = torchaudio.load(path)
hwav, sr = denoise(dwav[0], sr, args.run_dir, args.device)
out_path = args.out_dir / path.relative_to(args.in_dir)
out_path.parent.mkdir(parents=True, exist_ok=True)
torchaudio.save(out_path, hwav[None], sr)
if __name__ == "__main__":
main()
|