├── .gitignore ├── LICENSE ├── README.md ├── config ├── inference.toml └── train.toml ├── inference.sh ├── mertrics.sh ├── run.sh └── speech_enhance ├── __init__.py ├── audio_zen ├── __init__.py ├── acoustics │ ├── __init__.py │ ├── beamforming.py │ ├── feature.py │ ├── mask.py │ └── utils.py ├── constant.py ├── dataset │ ├── __init__.py │ └── base_dataset.py ├── inferencer │ ├── __init__.py │ └── base_inferencer.py ├── loss.py ├── metrics.py ├── model │ ├── __init__.py │ ├── base_model.py │ └── module │ │ ├── __init__.py │ │ ├── attention_model.py │ │ ├── causal_conv.py │ │ ├── feature_norm.py │ │ ├── sequence_model.py │ │ └── si_module.py ├── trainer │ ├── __init__.py │ └── base_trainer.py └── utils.py ├── fullsubnet ├── __init__.py ├── dataset │ ├── __init__.py │ ├── dataset_inference.py │ ├── dataset_train.py │ └── dataset_validation.py ├── inferencer │ ├── __init__.py │ └── inferencer.py ├── model │ ├── __init__.py │ └── fullsubnet.py └── trainer │ ├── __init__.py │ └── trainer.py ├── fullsubnet_plus ├── __init__.py ├── dataset │ ├── __init__.py │ ├── dataset_inference.py │ ├── dataset_train.py │ └── dataset_validation.py ├── inferencer │ ├── __init__.py │ └── inferencer.py ├── model │ └── fullsubnet_plus.py └── trainer │ ├── __init__.py │ └── trainer.py ├── inter_subnet ├── __init__.py ├── dataset │ ├── __init__.py │ ├── dataset_inference.py │ ├── dataset_train.py │ └── dataset_validation.py ├── inferencer │ ├── __init__.py │ └── inferencer.py ├── model │ ├── Inter_SubNet.py │ └── __init__.py └── trainer │ ├── __init__.py │ ├── joint_trainer.py │ └── trainer.py ├── subband_model ├── __init__.py ├── dataset │ ├── __init__.py │ ├── dataset_inference.py │ ├── dataset_train.py │ └── dataset_validation.py ├── inferencer │ ├── __init__.py │ └── inferencer.py ├── model │ ├── __init__.py │ └── subband_model.py └── trainer │ ├── __init__.py │ ├── joint_trainer.py │ └── trainer.py ├── tools ├── __init__.py ├── analyse.py ├── calculate_metrics.py ├── collect_lst.py ├── dns_mos.py ├── gen_lst.py ├── inference.py ├── noisyspeech_synthesizer.py ├── resample_dir.py └── train.py └── utils ├── __init__.py ├── logger.py ├── plot.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/README.md -------------------------------------------------------------------------------- /config/inference.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/config/inference.toml -------------------------------------------------------------------------------- /config/train.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/config/train.toml -------------------------------------------------------------------------------- /inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/inference.sh -------------------------------------------------------------------------------- /mertrics.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/mertrics.sh -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/run.sh -------------------------------------------------------------------------------- /speech_enhance/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /speech_enhance/audio_zen/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/audio_zen/acoustics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/audio_zen/acoustics/beamforming.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/acoustics/beamforming.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/acoustics/feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/acoustics/feature.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/acoustics/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/acoustics/mask.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/acoustics/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/acoustics/utils.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/constant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/constant.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/audio_zen/dataset/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/dataset/base_dataset.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/inferencer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/audio_zen/inferencer/base_inferencer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/inferencer/base_inferencer.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/loss.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/metrics.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/audio_zen/model/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/model/base_model.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/model/module/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/audio_zen/model/module/attention_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/model/module/attention_model.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/model/module/causal_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/model/module/causal_conv.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/model/module/feature_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/model/module/feature_norm.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/model/module/sequence_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/model/module/sequence_model.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/model/module/si_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/model/module/si_module.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/audio_zen/trainer/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/trainer/base_trainer.py -------------------------------------------------------------------------------- /speech_enhance/audio_zen/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/audio_zen/utils.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/fullsubnet/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/fullsubnet/dataset/dataset_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet/dataset/dataset_inference.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet/dataset/dataset_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet/dataset/dataset_train.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet/dataset/dataset_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet/dataset/dataset_validation.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet/inferencer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/fullsubnet/inferencer/inferencer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet/inferencer/inferencer.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/fullsubnet/model/fullsubnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet/model/fullsubnet.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/fullsubnet/trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet/trainer/trainer.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet_plus/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/fullsubnet_plus/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/fullsubnet_plus/dataset/dataset_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet_plus/dataset/dataset_inference.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet_plus/dataset/dataset_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet_plus/dataset/dataset_train.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet_plus/dataset/dataset_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet_plus/dataset/dataset_validation.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet_plus/inferencer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/fullsubnet_plus/inferencer/inferencer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet_plus/inferencer/inferencer.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet_plus/model/fullsubnet_plus.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet_plus/model/fullsubnet_plus.py -------------------------------------------------------------------------------- /speech_enhance/fullsubnet_plus/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/fullsubnet_plus/trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/fullsubnet_plus/trainer/trainer.py -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/dataset/dataset_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/inter_subnet/dataset/dataset_inference.py -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/dataset/dataset_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/inter_subnet/dataset/dataset_train.py -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/dataset/dataset_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/inter_subnet/dataset/dataset_validation.py -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/inferencer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/inferencer/inferencer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/inter_subnet/inferencer/inferencer.py -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/model/Inter_SubNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/inter_subnet/model/Inter_SubNet.py -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/trainer/joint_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/inter_subnet/trainer/joint_trainer.py -------------------------------------------------------------------------------- /speech_enhance/inter_subnet/trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/inter_subnet/trainer/trainer.py -------------------------------------------------------------------------------- /speech_enhance/subband_model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/subband_model/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/subband_model/dataset/dataset_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/subband_model/dataset/dataset_inference.py -------------------------------------------------------------------------------- /speech_enhance/subband_model/dataset/dataset_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/subband_model/dataset/dataset_train.py -------------------------------------------------------------------------------- /speech_enhance/subband_model/dataset/dataset_validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/subband_model/dataset/dataset_validation.py -------------------------------------------------------------------------------- /speech_enhance/subband_model/inferencer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/subband_model/inferencer/inferencer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/subband_model/inferencer/inferencer.py -------------------------------------------------------------------------------- /speech_enhance/subband_model/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/subband_model/model/subband_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/subband_model/model/subband_model.py -------------------------------------------------------------------------------- /speech_enhance/subband_model/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/subband_model/trainer/joint_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/subband_model/trainer/joint_trainer.py -------------------------------------------------------------------------------- /speech_enhance/subband_model/trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/subband_model/trainer/trainer.py -------------------------------------------------------------------------------- /speech_enhance/tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/tools/analyse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/tools/analyse.py -------------------------------------------------------------------------------- /speech_enhance/tools/calculate_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/tools/calculate_metrics.py -------------------------------------------------------------------------------- /speech_enhance/tools/collect_lst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/tools/collect_lst.py -------------------------------------------------------------------------------- /speech_enhance/tools/dns_mos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/tools/dns_mos.py -------------------------------------------------------------------------------- /speech_enhance/tools/gen_lst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/tools/gen_lst.py -------------------------------------------------------------------------------- /speech_enhance/tools/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/tools/inference.py -------------------------------------------------------------------------------- /speech_enhance/tools/noisyspeech_synthesizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/tools/noisyspeech_synthesizer.py -------------------------------------------------------------------------------- /speech_enhance/tools/resample_dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/tools/resample_dir.py -------------------------------------------------------------------------------- /speech_enhance/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/tools/train.py -------------------------------------------------------------------------------- /speech_enhance/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /speech_enhance/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/utils/logger.py -------------------------------------------------------------------------------- /speech_enhance/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/utils/plot.py -------------------------------------------------------------------------------- /speech_enhance/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RookieJunChen/Inter-SubNet/HEAD/speech_enhance/utils/utils.py --------------------------------------------------------------------------------