├── .gitignore ├── README.md ├── config ├── inference │ └── basic.json5 └── train │ ├── baseline_model.json5 │ ├── crn_baseline.json5 │ └── fullband_subband_narrowband_parallel_fusion_9Neighbors.json5 ├── dataset ├── __init__.py ├── dataset_enhancement.py ├── dataset_variable_length_input.py ├── stft_wav_dataset.py └── wav_dataset.py ├── inference.py ├── inferencer ├── __init__.py ├── base_inferencer.py ├── inferencer.py └── inferencer_full_band.py ├── model ├── __init__.py ├── batch_norm_default.py ├── crn.py ├── crn_in_paper_batch_norm.py ├── crn_in_paper_layer_norm.py ├── crn_module.py ├── default.py ├── default_group_norm.py ├── default_layer_norm.py ├── default_no_batch_norm.py └── loss.py ├── train.py ├── trainer ├── __init__.py ├── base_trainer.py ├── causal_trainer.py ├── trainer_fixed_length_input_full_band.py └── trainer_fixed_length_input_full_band_si_sdr_loss.py └── util ├── __init__.py ├── metrics.py ├── utils.py └── visualization.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/README.md -------------------------------------------------------------------------------- /config/inference/basic.json5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/config/inference/basic.json5 -------------------------------------------------------------------------------- /config/train/baseline_model.json5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/config/train/baseline_model.json5 -------------------------------------------------------------------------------- /config/train/crn_baseline.json5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/config/train/crn_baseline.json5 -------------------------------------------------------------------------------- /config/train/fullband_subband_narrowband_parallel_fusion_9Neighbors.json5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/config/train/fullband_subband_narrowband_parallel_fusion_9Neighbors.json5 -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dataset/dataset_enhancement.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/dataset/dataset_enhancement.py -------------------------------------------------------------------------------- /dataset/dataset_variable_length_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/dataset/dataset_variable_length_input.py -------------------------------------------------------------------------------- /dataset/stft_wav_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/dataset/stft_wav_dataset.py -------------------------------------------------------------------------------- /dataset/wav_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/dataset/wav_dataset.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/inference.py -------------------------------------------------------------------------------- /inferencer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /inferencer/base_inferencer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/inferencer/base_inferencer.py -------------------------------------------------------------------------------- /inferencer/inferencer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/inferencer/inferencer.py -------------------------------------------------------------------------------- /inferencer/inferencer_full_band.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/inferencer/inferencer_full_band.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/batch_norm_default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/model/batch_norm_default.py -------------------------------------------------------------------------------- /model/crn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/model/crn.py -------------------------------------------------------------------------------- /model/crn_in_paper_batch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/model/crn_in_paper_batch_norm.py -------------------------------------------------------------------------------- /model/crn_in_paper_layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/model/crn_in_paper_layer_norm.py -------------------------------------------------------------------------------- /model/crn_module.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/model/default.py -------------------------------------------------------------------------------- /model/default_group_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/model/default_group_norm.py -------------------------------------------------------------------------------- /model/default_layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/model/default_layer_norm.py -------------------------------------------------------------------------------- /model/default_no_batch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/model/default_no_batch_norm.py -------------------------------------------------------------------------------- /model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/model/loss.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/train.py -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /trainer/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/trainer/base_trainer.py -------------------------------------------------------------------------------- /trainer/causal_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/trainer/causal_trainer.py -------------------------------------------------------------------------------- /trainer/trainer_fixed_length_input_full_band.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/trainer/trainer_fixed_length_input_full_band.py -------------------------------------------------------------------------------- /trainer/trainer_fixed_length_input_full_band_si_sdr_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/trainer/trainer_fixed_length_input_full_band_si_sdr_loss.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /util/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/util/metrics.py -------------------------------------------------------------------------------- /util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/util/utils.py -------------------------------------------------------------------------------- /util/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haoxiangsnr/A-Convolutional-Recurrent-Neural-Network-for-Real-Time-Speech-Enhancement/HEAD/util/visualization.py --------------------------------------------------------------------------------