BS-RoFormer wind stem for slurper, fp16 Core ML with golden chunk
Browse files
.gitattributes
CHANGED
|
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
golden_horns.f32 filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
golden_raw.f32 filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: coreml
|
| 3 |
+
tags:
|
| 4 |
+
- audio
|
| 5 |
+
- music-source-separation
|
| 6 |
+
- bs-roformer
|
| 7 |
+
- coreml
|
| 8 |
+
base_model: noblebarkrr/BS-Roformer-MVSep-Mega-53-stems
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# BSRoformer-Wind-CoreML
|
| 12 |
+
|
| 13 |
+
The wind stem (brass and woodwinds) of the [MVSep Mega 53-stem BS-RoFormer](https://github.com/ZFTurbo/Music-Source-Separation-Training/releases/tag/v1.0.21), converted to Core ML for [slurper](https://github.com/trevors/slurper), where it produces the "horns" stem.
|
| 14 |
+
|
| 15 |
+
- `bsr_wind_fp16.mlpackage`: the graph `frames[1,2,690,2048] -> recon[1,2,690,2048]` for one 8 s chunk at 44.1 kHz, framed with a 2048-sample periodic Hann window at hop 512: windowed DFT, band split, the axial rotary transformer, mask estimator, complex mask multiply and inverse DFT, all inside the model in float16. The host reflect-pads and frames the chunk and overlap-adds the output divided by the summed squared window.
|
| 16 |
+
- `golden_raw.f32`, `golden_horns.f32`: an 8 s stereo chunk (channel-major float32) and PyTorch's output for it, used by slurper's tests. Core ML matches PyTorch on it at cosine 0.99998 (44.7 dB SDR).
|
| 17 |
+
|
| 18 |
+
Produced by `scripts/convert_bs_roformer.py` in the slurper repository from the single-stem repack [noblebarkrr/BS-Roformer-MVSep-Mega-53-stems](https://huggingface.co/noblebarkrr/BS-Roformer-MVSep-Mega-53-stems) (`v1/bs_mega_53stem_wind_mvsep.ckpt`), with the model code from [Music-Source-Separation-Training](https://github.com/ZFTurbo/Music-Source-Separation-Training) (MIT). See the upstream release for the weights' terms.
|
| 19 |
+
|
| 20 |
+
The golden chunk is 8 s from 0:32 of "Blues for Mundy" by The Airmen of Note, The United States Air Force Band (composed by Rick Whitehead), from *60 Years of the Airmen of Note* (2011), via [Wikimedia Commons](https://commons.wikimedia.org/wiki/File:Blues_for_Mundy_-_Airmen_of_Note_-_United_States_Air_Force_Band.mp3). As a work of the US federal government the recording is in the public domain.
|
bsr_wind_fp16.mlpackage/Data/com.apple.CoreML/model.mlmodel
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dbb26610e43376763231deaf8a52f3e296ce50213dd6bb8ac1a452c6f2c70e1e
|
| 3 |
+
size 748990
|
bsr_wind_fp16.mlpackage/Data/com.apple.CoreML/weights/weight.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:acc4dc39bc044f547da86ac399300fdf41f9fed5c114e901a0766b7c5e36fb32
|
| 3 |
+
size 94421760
|
bsr_wind_fp16.mlpackage/Manifest.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"fileFormatVersion": "1.0.0",
|
| 3 |
+
"itemInfoEntries": {
|
| 4 |
+
"B0268636-0C8F-41A3-9A12-8E5E4EB52A7F": {
|
| 5 |
+
"author": "com.apple.CoreML",
|
| 6 |
+
"description": "CoreML Model Weights",
|
| 7 |
+
"name": "weights",
|
| 8 |
+
"path": "com.apple.CoreML/weights"
|
| 9 |
+
},
|
| 10 |
+
"F5730EEA-E431-4FBB-B005-B973B8041A83": {
|
| 11 |
+
"author": "com.apple.CoreML",
|
| 12 |
+
"description": "CoreML Model Specification",
|
| 13 |
+
"name": "model.mlmodel",
|
| 14 |
+
"path": "com.apple.CoreML/model.mlmodel"
|
| 15 |
+
}
|
| 16 |
+
},
|
| 17 |
+
"rootModelIdentifier": "F5730EEA-E431-4FBB-B005-B973B8041A83"
|
| 18 |
+
}
|
golden_horns.f32
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0f5ed9182b7d87deaabaa778cbbe7a42e68ac0d1379b0bcad50b436f88973eef
|
| 3 |
+
size 2822400
|
golden_raw.f32
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4e91dd446ccc25ce3c597d4aa8dee7f5e40ee76afcb08e56d0a07bca750ca37f
|
| 3 |
+
size 2822400
|