├── .github └── workflows │ └── jekyll-gh-pages.yml ├── .gitignore ├── McNetCLI.py ├── README.md ├── config ├── mc_net_offline.yaml └── mc_net_online.yaml ├── examples ├── README.md └── mix.png ├── model_checkpoints ├── offline │ └── epoch494_criteria18.78_sdr18.78.ckpt └── online │ └── epoch454_criteria18.35_sdr18.35.ckpt ├── requirements.txt └── src ├── __init__.py ├── dataset ├── __init__.py ├── inf_dataset.py ├── my_data_module.py ├── predict_dataset.py ├── train_dataset.py └── val_dataset.py ├── model ├── __init__.py └── mc_net.py └── util ├── __init__.py ├── acoustic_utils.py ├── cal_metrics.py ├── git_tools.py ├── lit_progress_bar.py ├── metrics.py ├── move_log_to_dir.py ├── my_json_encoder.py ├── my_logger.py ├── my_rich_progress_bar.py ├── my_save_config_callback.py ├── utils.py └── visualization.py /.github/workflows/jekyll-gh-pages.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/.github/workflows/jekyll-gh-pages.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/.gitignore -------------------------------------------------------------------------------- /McNetCLI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/McNetCLI.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/README.md -------------------------------------------------------------------------------- /config/mc_net_offline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/config/mc_net_offline.yaml -------------------------------------------------------------------------------- /config/mc_net_online.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/config/mc_net_online.yaml -------------------------------------------------------------------------------- /examples/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/examples/README.md -------------------------------------------------------------------------------- /examples/mix.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/examples/mix.png -------------------------------------------------------------------------------- /model_checkpoints/offline/epoch494_criteria18.78_sdr18.78.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/model_checkpoints/offline/epoch494_criteria18.78_sdr18.78.ckpt -------------------------------------------------------------------------------- /model_checkpoints/online/epoch454_criteria18.35_sdr18.35.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/model_checkpoints/online/epoch454_criteria18.35_sdr18.35.ckpt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/dataset/inf_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/dataset/inf_dataset.py -------------------------------------------------------------------------------- /src/dataset/my_data_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/dataset/my_data_module.py -------------------------------------------------------------------------------- /src/dataset/predict_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/dataset/predict_dataset.py -------------------------------------------------------------------------------- /src/dataset/train_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/dataset/train_dataset.py -------------------------------------------------------------------------------- /src/dataset/val_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/dataset/val_dataset.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/model/mc_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/model/mc_net.py -------------------------------------------------------------------------------- /src/util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/util/acoustic_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/acoustic_utils.py -------------------------------------------------------------------------------- /src/util/cal_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/cal_metrics.py -------------------------------------------------------------------------------- /src/util/git_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/git_tools.py -------------------------------------------------------------------------------- /src/util/lit_progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/lit_progress_bar.py -------------------------------------------------------------------------------- /src/util/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/metrics.py -------------------------------------------------------------------------------- /src/util/move_log_to_dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/move_log_to_dir.py -------------------------------------------------------------------------------- /src/util/my_json_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/my_json_encoder.py -------------------------------------------------------------------------------- /src/util/my_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/my_logger.py -------------------------------------------------------------------------------- /src/util/my_rich_progress_bar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/my_rich_progress_bar.py -------------------------------------------------------------------------------- /src/util/my_save_config_callback.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/my_save_config_callback.py -------------------------------------------------------------------------------- /src/util/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/utils.py -------------------------------------------------------------------------------- /src/util/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Audio-WestlakeU/McNet/HEAD/src/util/visualization.py --------------------------------------------------------------------------------