drumsep-onnx / README.md
philippherzig's picture
Move from philippherzig to gridshiftstudio org
740429a verified
---
license: mit
tags:
- audio
- source-separation
- drumsep
- onnx
- onnxruntime
library_name: onnx
---
# DrumSep (ONNX) — Gridshift mirror
Drum-element separation model (kick / snare / cymbals / toms) in ONNX format,
moved from `philippherzig/drumsep-onnx` to
[Gridshift Studio](https://huggingface.co/gridshiftstudio) org for
release-pipeline consistency.
Weights and format are unchanged from the previous repository. The original
upstream is [inagoy/drumsep](https://github.com/inagoy/drumsep),
a DemucsHT fine-tune on drum stems.
## Usage
Consumed by [Gridshift](https://gridshift.studio)'s drum-element separation
feature via the Rust stem-splitter crate. The `drumsep_manifest.json` in this
repo lists the ONNX model file with SHA-256 for integrity verification.
- **Audio format:** 44.1 kHz stereo, 1 764 000 samples per inference window
- **Stems:** kick, snare, cymbals, toms
- **Format:** ONNX (opset 14)
- **File size:** ~335 MB
## License and attribution
Inherited MIT license from the upstream `inagoy/drumsep` project. The ONNX
conversion was done from the original HDemucs checkpoint using
`torch.onnx.export` (see `app/rust/drum-separation/convert_drumsep_to_onnx.py`
in the Gridshift source tree).