Spaces:

akhaliq
/

Music_Source_Separation

Runtime error

Music_Source_Separation / separate_scripts /separate_accompaniment.sh

akhaliq3

spaces demo

5019931 over 2 years ago

No virus

1.02 kB

	#!/bin/bash
	AUDIO_PATH=${1:-"./resources/vocals_accompaniment_10s.mp3"} # The path of audio to be separated.
	OUTPUT_PATH=${2:-"./sep_results/sep_vocals.mp3"} # The path to write out separated audio.

	MODEL_NAME="resunet_subbandtime" # "resunet_ismir2021" \| ""resunet_subbandtime""

	if [ $MODEL_NAME = "resunet_ismir2021" ]; then
	CHECKPOINT_PATH="./downloaded_checkpoints/resunet143_ismir2021_accompaniment_16.8dB_350k_steps.pth"
	TRAIN_CONFIG_YAML="./scripts/4_train/musdb18/configs/accompaniment-vocals,resunet_ismir2021.yaml"

	elif [ $MODEL_NAME = "resunet_subbandtime" ]; then
	TRAIN_CONFIG_YAML="./scripts/4_train/musdb18/configs/accompaniment-vocals,resunet_subbandtime.yaml"
	CHECKPOINT_PATH="./downloaded_checkpoints/resunet143_subbtandtime_accompaniment_16.4dB_350k_steps.pth"
	fi

	# Inference
	CUDA_VISIBLE_DEVICES=0 python3 bytesep/inference.py \
	--config_yaml=$TRAIN_CONFIG_YAML \
	--checkpoint_path=$CHECKPOINT_PATH \
	--audio_path=$AUDIO_PATH \
	--output_path=$OUTPUT_PATH