Model conversion: TF to TF.js

sigsep / open-unmix-js

web based audio unmixing

https://sigsep.github.io/js

MIT License

12 stars 1 forks source link

Model conversion: TF to TF.js #3

Closed faroit closed 4 years ago

faroit commented 4 years ago

write a python script that lives in a subfolder called "models"

[x] download the 2-stem model that can be downloaded here
[x] convert this to a tfjs model
[x] apply compression if suitable

do not forget to add (conda) environment files.

faroit commented 4 years ago

conversion is not necessary anymore: see this repo

faroit commented 4 years ago

Please check if these models work fine. If yes, wen can close this issue

ClaraJacintho commented 4 years ago

We managed to make the model work. We still haven't tested it with real data, but we were able to load the model and make a prediction using dummy values.

It currently only has the 2-stem model though. Should we close the issue now or wait for the 4-stem model?

faroit commented 4 years ago

👌

The 2-stem model is totally fine. The 4-stem model will consume too much memory in the browser anyways