|
#!/bin/bash |
|
AUDIO_PATH=${1:-"./resources/vocals_accompaniment_10s.mp3"} |
|
OUTPUT_PATH=${2:-"./sep_results/sep_vocals.mp3"} |
|
|
|
MODEL_NAME="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 |
|
|
|
|
|
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 |
|
|