stem-splitter
BS-RoFormerを使用して音声を6ステム(bass、drums、other、vocals、guitar、piano)に分離するPythonパッケージです。
CUDA推論ではFP16 mixed precisionとTransformer層のtorch.compileを既定で使用します。compileを使用できない環境では警告を出してeager実行へフォールバックします。Windowsでcompileを使用する場合は、PyTorchのバージョンに対応するtriton-windowsを別途インストールしてください。
初回推論ではGPU向けカーネルを生成するため、数十秒程度のcompile待ちが発生します。生成後のカーネルはキャッシュされます。
pip install stem-splitter
stem-splitter input.wav --out-dir output
compileを明示的に無効化する場合は--no-compile、CUDAでFP32推論を行う場合は--full-precisionを指定します。
未ラベル蒸留の検証
teacher pseudo-labelによる小型BS-RoFormerの蒸留・弱stemアブレーションは、ライブラリ本体を変更せず distillation/ に分離しています。実測結果と採否判断は distillation/ABLATION_REPORT.md を参照してください。
Metadata
Release files for stem-splitter 0.0.8
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| stem_splitter-0.0.8.tar.gz | 22.9 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| stem_splitter-0.0.8-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 44.3 kB
Release files / stem_splitter-0.0.8.tar.gz
| Download URL | stem_splitter-0.0.8.tar.gz |
|---|---|
| Size | 22.9 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
a1c37d8d69719118c3ec3c6b2d95ba106ef50185021b2fbf9468adbfc91f79de
|
|
BLAKE2b-256 checksum How to use checksums |
65d3ad1996a30a1b08ccec1e2884f3e26f7d260a75aa9e32d5a8c8782f400ad3
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.10.6
|
Release files / stem_splitter-0.0.8-py3-none-any.whl
| Download URL | stem_splitter-0.0.8-py3-none-any.whl |
|---|---|
| Size | 21.4 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
d843e47ea4d15e431fb6cf80f3cdd534a2ebc16b4ac1a6672732bff6490200a9
|
|
BLAKE2b-256 checksum How to use checksums |
13d1e865fc89f11db4fcd8bd3d2c840d2450d72d6a9c5ff616966ea7ac9d49ff
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.10.6
|