├── .flake8 ├── .github └── workflows │ ├── lint.yml │ └── pythonpackage.yml ├── .gitignore ├── CHANGELOG.rst ├── LICENSE ├── README.rst ├── docs ├── Makefile ├── bf.rst ├── bss.rst ├── conf.py ├── derev.rst ├── index.rst ├── make.bat ├── modules.rst ├── requirements.txt ├── setup.rst ├── torchiva.models.rst └── torchiva.rst ├── environment.yml ├── examples ├── autoclip_module.py ├── configs │ ├── gev.json │ ├── mvdr.json │ ├── mwf.json │ └── tiss.json ├── data_module.py ├── dataloader.py ├── example.py ├── example_dnn.py ├── export_model.py ├── requirements.txt ├── samples │ ├── README.md │ ├── dry │ │ ├── 103-1240-0003_1235-135887-0017_s1.wav │ │ └── 103-1240-0003_1235-135887-0017_s2.wav │ ├── make_mix.py │ ├── mix_reverb │ │ └── 103-1240-0003_1235-135887-0017.wav │ ├── read_samples.py │ ├── ref_anecho │ │ ├── 103-1240-0003_1235-135887-0017_s1.wav │ │ └── 103-1240-0003_1235-135887-0017_s2.wav │ ├── ref_reverb │ │ ├── 103-1240-0003_1235-135887-0017_s1.wav │ │ └── 103-1240-0003_1235-135887-0017_s2.wav │ └── samples_list.yaml ├── separation_model.py ├── test.py └── train.py ├── pyproject.yaml ├── setup.cfg ├── setup.py ├── tests ├── requirements.txt ├── test_autograd.py ├── test_beamformer.py ├── test_dtypes.py ├── test_fftconvolve.py ├── test_iss_t_rev.py ├── test_linalg.py ├── test_loader.py ├── test_overtiss.py ├── test_separate.py ├── test_separate_command.sh └── test_wpe.py ├── torchiva ├── __init__.py ├── auxiva_ip.py ├── auxiva_ip2.py ├── base.py ├── beamformer.py ├── dtypes.py ├── fftconvolve.py ├── five.py ├── linalg.py ├── loader.py ├── models │ ├── __init__.py │ ├── blstm.py │ ├── classic.py │ ├── glu.py │ ├── parameters.py │ └── simple.py ├── nn.py ├── parameters.py ├── scaling.py ├── separate.py ├── stft.py ├── t_iss.py ├── utils.py └── wpe.py └── trained_models └── tiss ├── model_config.yaml └── model_weights.ckpt /.flake8: -------------------------------------------------------------------------------- 1 | [flake8] 2 | max-line-length = 88 3 | -------------------------------------------------------------------------------- /.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/.github/workflows/lint.yml -------------------------------------------------------------------------------- /.github/workflows/pythonpackage.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/.github/workflows/pythonpackage.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/CHANGELOG.rst -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/LICENSE -------------------------------------------------------------------------------- /README.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/README.rst -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/bf.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/docs/bf.rst -------------------------------------------------------------------------------- /docs/bss.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/docs/bss.rst -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/derev.rst: -------------------------------------------------------------------------------- 1 | WPE 2 | ~~~ 3 | .. autoclass:: torchiva.WPE -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/docs/make.bat -------------------------------------------------------------------------------- /docs/modules.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/docs/modules.rst -------------------------------------------------------------------------------- /docs/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/docs/requirements.txt -------------------------------------------------------------------------------- /docs/setup.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/docs/setup.rst -------------------------------------------------------------------------------- /docs/torchiva.models.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/docs/torchiva.models.rst -------------------------------------------------------------------------------- /docs/torchiva.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/docs/torchiva.rst -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/environment.yml -------------------------------------------------------------------------------- /examples/autoclip_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/autoclip_module.py -------------------------------------------------------------------------------- /examples/configs/gev.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/configs/gev.json -------------------------------------------------------------------------------- /examples/configs/mvdr.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/configs/mvdr.json -------------------------------------------------------------------------------- /examples/configs/mwf.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/configs/mwf.json -------------------------------------------------------------------------------- /examples/configs/tiss.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/configs/tiss.json -------------------------------------------------------------------------------- /examples/data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/data_module.py -------------------------------------------------------------------------------- /examples/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/dataloader.py -------------------------------------------------------------------------------- /examples/example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/example.py -------------------------------------------------------------------------------- /examples/example_dnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/example_dnn.py -------------------------------------------------------------------------------- /examples/export_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/export_model.py -------------------------------------------------------------------------------- /examples/requirements.txt: -------------------------------------------------------------------------------- 1 | pytorch_lightning -------------------------------------------------------------------------------- /examples/samples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/samples/README.md -------------------------------------------------------------------------------- /examples/samples/dry/103-1240-0003_1235-135887-0017_s1.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/samples/dry/103-1240-0003_1235-135887-0017_s1.wav -------------------------------------------------------------------------------- /examples/samples/dry/103-1240-0003_1235-135887-0017_s2.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/samples/dry/103-1240-0003_1235-135887-0017_s2.wav -------------------------------------------------------------------------------- /examples/samples/make_mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/samples/make_mix.py -------------------------------------------------------------------------------- /examples/samples/mix_reverb/103-1240-0003_1235-135887-0017.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/samples/mix_reverb/103-1240-0003_1235-135887-0017.wav -------------------------------------------------------------------------------- /examples/samples/read_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/samples/read_samples.py -------------------------------------------------------------------------------- /examples/samples/ref_anecho/103-1240-0003_1235-135887-0017_s1.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/samples/ref_anecho/103-1240-0003_1235-135887-0017_s1.wav -------------------------------------------------------------------------------- /examples/samples/ref_anecho/103-1240-0003_1235-135887-0017_s2.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/samples/ref_anecho/103-1240-0003_1235-135887-0017_s2.wav -------------------------------------------------------------------------------- /examples/samples/ref_reverb/103-1240-0003_1235-135887-0017_s1.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/samples/ref_reverb/103-1240-0003_1235-135887-0017_s1.wav -------------------------------------------------------------------------------- /examples/samples/ref_reverb/103-1240-0003_1235-135887-0017_s2.wav: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/samples/ref_reverb/103-1240-0003_1235-135887-0017_s2.wav -------------------------------------------------------------------------------- /examples/samples/samples_list.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/samples/samples_list.yaml -------------------------------------------------------------------------------- /examples/separation_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/separation_model.py -------------------------------------------------------------------------------- /examples/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/test.py -------------------------------------------------------------------------------- /examples/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/examples/train.py -------------------------------------------------------------------------------- /pyproject.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/pyproject.yaml -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/setup.py -------------------------------------------------------------------------------- /tests/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/requirements.txt -------------------------------------------------------------------------------- /tests/test_autograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/test_autograd.py -------------------------------------------------------------------------------- /tests/test_beamformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/test_beamformer.py -------------------------------------------------------------------------------- /tests/test_dtypes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/test_dtypes.py -------------------------------------------------------------------------------- /tests/test_fftconvolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/test_fftconvolve.py -------------------------------------------------------------------------------- /tests/test_iss_t_rev.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/test_iss_t_rev.py -------------------------------------------------------------------------------- /tests/test_linalg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/test_linalg.py -------------------------------------------------------------------------------- /tests/test_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/test_loader.py -------------------------------------------------------------------------------- /tests/test_overtiss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/test_overtiss.py -------------------------------------------------------------------------------- /tests/test_separate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/test_separate.py -------------------------------------------------------------------------------- /tests/test_separate_command.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/test_separate_command.sh -------------------------------------------------------------------------------- /tests/test_wpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/tests/test_wpe.py -------------------------------------------------------------------------------- /torchiva/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/__init__.py -------------------------------------------------------------------------------- /torchiva/auxiva_ip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/auxiva_ip.py -------------------------------------------------------------------------------- /torchiva/auxiva_ip2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/auxiva_ip2.py -------------------------------------------------------------------------------- /torchiva/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/base.py -------------------------------------------------------------------------------- /torchiva/beamformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/beamformer.py -------------------------------------------------------------------------------- /torchiva/dtypes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/dtypes.py -------------------------------------------------------------------------------- /torchiva/fftconvolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/fftconvolve.py -------------------------------------------------------------------------------- /torchiva/five.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/five.py -------------------------------------------------------------------------------- /torchiva/linalg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/linalg.py -------------------------------------------------------------------------------- /torchiva/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/loader.py -------------------------------------------------------------------------------- /torchiva/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/models/__init__.py -------------------------------------------------------------------------------- /torchiva/models/blstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/models/blstm.py -------------------------------------------------------------------------------- /torchiva/models/classic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/models/classic.py -------------------------------------------------------------------------------- /torchiva/models/glu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/models/glu.py -------------------------------------------------------------------------------- /torchiva/models/parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/models/parameters.py -------------------------------------------------------------------------------- /torchiva/models/simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/models/simple.py -------------------------------------------------------------------------------- /torchiva/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/nn.py -------------------------------------------------------------------------------- /torchiva/parameters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/parameters.py -------------------------------------------------------------------------------- /torchiva/scaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/scaling.py -------------------------------------------------------------------------------- /torchiva/separate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/separate.py -------------------------------------------------------------------------------- /torchiva/stft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/stft.py -------------------------------------------------------------------------------- /torchiva/t_iss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/t_iss.py -------------------------------------------------------------------------------- /torchiva/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/utils.py -------------------------------------------------------------------------------- /torchiva/wpe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/torchiva/wpe.py -------------------------------------------------------------------------------- /trained_models/tiss/model_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/trained_models/tiss/model_config.yaml -------------------------------------------------------------------------------- /trained_models/tiss/model_weights.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fakufaku/torchiva/HEAD/trained_models/tiss/model_weights.ckpt --------------------------------------------------------------------------------