diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000..50cdfa9 --- /dev/null +++ b/.gitattributes @@ -0,0 +1 @@ +.ipynb linguist-documentation diff --git a/MANIFEST.in b/MANIFEST.in new file mode 100644 index 0000000..2bc28e7 --- /dev/null +++ b/MANIFEST.in @@ -0,0 +1,4 @@ +recursive-include mayavoz *.py +recursive-include mayavoz *.yaml +global-exclude *.pyc +global-exclude __pycache__ diff --git a/README.md b/README.md index c7fc2db..20cf995 100644 --- a/README.md +++ b/README.md @@ -20,7 +20,7 @@ mayavoz is a Pytorch-based opensource toolkit for speech enhancement. It is desi ## Quick Start :fire: ``` python -from mayavoz import Mayamodel +from mayavoz.models import Mayamodel model = Mayamodel.from_pretrained("shahules786/mayavoz-waveunet-valentini-28spk") model("noisy_audio.wav") @@ -33,7 +33,9 @@ model("noisy_audio.wav") | WaveUnet | Vctk-28spk | 0.836 | 2.78 | shahules786/mayavoz-waveunet-valentini-28spk | | Demucs | Vctk-28spk | 0.961 | 2.56 | shahules786/mayavoz-demucs-valentini-28spk | | DCCRN | Vctk-28spk | 0.724 | 2.55 | shahules786/mayavoz-dccrn-valentini-28spk | +| Demucs | DNS2020 (20hrs) | 0.56 | 1.26 | shahules786/mayavoz-demucs-dns2020-20hr | +Test scores are based on respective test set associated with train dataset. ## Installation Only Python 3.8+ is officially supported (though it might work with Python 3.7) @@ -56,3 +58,10 @@ git clone url cd mayavoz pip install -e . ``` + +## Support + +For commercial enquiries and scientific consulting, please [contact me](https://shahules786.github.io/). + +### Acknowledgements +Sincere gratitude to [AMPLYFI](https://amplyfi.com/) for supporting this project. diff --git a/requirements.txt b/requirements.txt index fb54920..d046883 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,7 +3,7 @@ huggingface-hub>=0.10.0 hydra-core>=1.2.0 joblib>=1.2.0 librosa>=0.9.2 -mlflow>=1.29.0 +mlflow>=1.28.0 numpy>=1.23.3 pesq==0.0.4 protobuf>=3.19.6 diff --git a/setup.cfg b/setup.cfg index b31929e..da88a56 100644 --- a/setup.cfg +++ b/setup.cfg @@ -98,3 +98,7 @@ exclude = build dist .eggs + +[options.data_files] +. = requirements.txt +_ = version.txt