Export spleeter model to onnx for source separation (#2237)
This commit is contained in:
41
scripts/spleeter/run.sh
Executable file
41
scripts/spleeter/run.sh
Executable file
@@ -0,0 +1,41 @@
|
||||
#!/usr/bin/env bash
|
||||
|
||||
|
||||
if [ ! -f 2stems.tar.gz ]; then
|
||||
curl -SL -O https://github.com/deezer/spleeter/releases/download/v1.4.0/2stems.tar.gz
|
||||
fi
|
||||
|
||||
if [ ! -d ./2stems ]; then
|
||||
mkdir -p 2stems
|
||||
cd 2stems
|
||||
tar xvf ../2stems.tar.gz
|
||||
cd ..
|
||||
fi
|
||||
|
||||
ls -lh
|
||||
|
||||
ls -lh 2stems
|
||||
|
||||
if [ ! -f 2stems/frozen_vocals_model.pb ]; then
|
||||
python3 ./convert_to_pb.py \
|
||||
--model-dir ./2stems \
|
||||
--output-node-names vocals_spectrogram/mul \
|
||||
--output-filename ./2stems/frozen_vocals_model.pb
|
||||
fi
|
||||
|
||||
ls -lh 2stems
|
||||
|
||||
if [ ! -f 2stems/frozen_accompaniment_model.pb ]; then
|
||||
python3 ./convert_to_pb.py \
|
||||
--model-dir ./2stems \
|
||||
--output-node-names accompaniment_spectrogram/mul \
|
||||
--output-filename ./2stems/frozen_accompaniment_model.pb
|
||||
fi
|
||||
|
||||
ls -lh 2stems
|
||||
|
||||
python3 ./convert_to_torch.py --name vocals
|
||||
python3 ./convert_to_torch.py --name accompaniment
|
||||
python3 ./export_onnx.py
|
||||
|
||||
ls -lh 2stems
|
||||
Reference in New Issue
Block a user