Add UVR models for source separation. (#2266)

This commit is contained in:
Fangjun Kuang
2025-05-31 13:31:31 +08:00
committed by GitHub
parent 93e2819c18
commit 921f0f40cb
7 changed files with 575 additions and 3 deletions

54
scripts/uvr_mdx/run.sh Executable file
View File

@@ -0,0 +1,54 @@
#!/usr/bin/env bash
# Copyright 2025 Xiaomi Corp. (authors: Fangjun Kuang)
set -ex
# Please see https://github.com/TRvlvr/model_repo/releases/tag/all_public_uvr_models
models=(
UVR-MDX-NET-Inst_1.onnx
UVR-MDX-NET-Inst_2.onnx
UVR-MDX-NET-Inst_3.onnx
UVR-MDX-NET-Inst_HQ_1.onnx
UVR-MDX-NET-Inst_HQ_2.onnx
UVR-MDX-NET-Inst_HQ_3.onnx
UVR-MDX-NET-Inst_HQ_4.onnx
UVR-MDX-NET-Inst_HQ_5.onnx
UVR-MDX-NET-Inst_Main.onnx
UVR-MDX-NET-Voc_FT.onnx
UVR-MDX-NET_Crowd_HQ_1.onnx
UVR_MDXNET_1_9703.onnx
UVR_MDXNET_2_9682.onnx
UVR_MDXNET_3_9662.onnx
UVR_MDXNET_9482.onnx
UVR_MDXNET_KARA.onnx
UVR_MDXNET_KARA_2.onnx
UVR_MDXNET_Main.onnx
)
mkdir -p models
for m in ${models[@]}; do
if [ ! -f models/$m ]; then
curl -SL --output models/$m https://github.com/TRvlvr/model_repo/releases/download/all_public_uvr_models/$m
fi
done
ls -lh models
for m in ${models[@]}; do
echo "----------$m----------"
python3 ./add_meta_data_and_quantize.py --filename models/$m
ls -lh models/
done
if [ -f ./audio_example.wav ]; then
for m in ${models[@]}; do
./test.py --model-filename ./models/$m --audio-filename ./audio_example.wav
name=$(basename -s .onnx $m)
mv -v vocals.mp3 ${name}_vocals.mp3
mv -v non_vocals.mp3 ${name}_non_vocals.mp3
done
ls -lh *.mp3
fi