generated from thinkode/modelRepository
Initial commit and v1.0
This commit is contained in:
46
audioSeparator.py
Normal file
46
audioSeparator.py
Normal file
@@ -0,0 +1,46 @@
|
||||
import demucs.separate
|
||||
import shutil
|
||||
import argparse
|
||||
import os
|
||||
import shutil
|
||||
import json
|
||||
from log import log_step
|
||||
from genericpath import exists
|
||||
|
||||
try:
|
||||
# Create the parser
|
||||
parser = argparse.ArgumentParser(description="dubstudio audio separator tool (v1.0) - V. BOULANGER - 2025")
|
||||
|
||||
# Options definition
|
||||
parser.add_argument('-a', '--audio', help='Audio file to process', default='audio.wav')
|
||||
parser.add_argument('-f', '--folder', help="Output folder name where the WAV audio files will be stored", default='artifacts')
|
||||
parser.add_argument('-m', '--model', help="Separation model to use for processing the audio", default='htdemucs')
|
||||
|
||||
# Options analyzing
|
||||
args = parser.parse_args()
|
||||
|
||||
log_step('init', 100, json.dumps({
|
||||
"audioFile": args.audio,
|
||||
"outputFolder": args.folder,
|
||||
"model": args.model,
|
||||
}))
|
||||
|
||||
# Separate vocals with progress
|
||||
demucs.separate.main([
|
||||
"--two-stems", "vocals",
|
||||
"--filename", "{stem}.wav",
|
||||
"-n", args.model,
|
||||
"--float32",
|
||||
"--out", args.folder,
|
||||
args.audio
|
||||
])
|
||||
|
||||
except KeyboardInterrupt:
|
||||
# Delete the output folder
|
||||
if exists(args.folder):
|
||||
shutil.rmtree(args.folder)
|
||||
log_step("exit", 100, "program exit")
|
||||
|
||||
except Exception as e:
|
||||
log_step("error", 100, str(e))
|
||||
|
||||
Reference in New Issue
Block a user