Add C++ runtime for spleeter about source separation (#2242)

2025-05-23 22:30:57 +08:00
parent 55a44793e6
commit 716ba8317b
28 changed files with 1267 additions and 72 deletions
--- a/scripts/spleeter/convert_to_torch.py
+++ b/scripts/spleeter/convert_to_torch.py
@@ -217,8 +217,8 @@ def main(name):
    # for the batchnormalization in torch,
    # default input shape is NCHW

-    # NHWC to NCHW
-    torch_y1_out = unet(torch.from_numpy(y0_out).permute(0, 3, 1, 2))
+    torch_y1_out = unet(torch.from_numpy(y0_out).permute(3, 0, 1, 2))
+    torch_y1_out = torch_y1_out.permute(1, 0, 2, 3)

    #  print(torch_y1_out.shape, torch.from_numpy(y1_out).permute(0, 3, 1, 2).shape)
    assert torch.allclose(