├── LICENSE ├── README.md ├── __config__.py ├── images ├── Selection_059.png ├── Selection_060.png ├── Selection_061.png ├── Selection_062.png └── fuss_results.png ├── pretrained_models ├── .gitattributes └── download_pretrained_models.sh ├── requirements.txt └── sudo_rm_rf ├── ._.DS_Store ├── __init__.py ├── dnn ├── __init__.py ├── dataset_loader │ ├── __init__.py │ ├── abstract_dataset.py │ ├── augmented_mix_dataloader.py │ ├── fuss.py │ ├── libri2mix.py │ ├── musdb_dataset.py │ ├── torch_dataloader.py │ ├── wham.py │ └── whamr.py ├── experiments │ ├── __init__.py │ ├── baseline_runner.py │ ├── run_attentive_sudormrf.py │ ├── run_fuss_separation.py │ ├── run_improved_sudormrf.py │ ├── run_improved_sudormrf_reverb.py │ ├── run_sudormrf.py │ ├── run_sudormrf_gc_v2.py │ └── utils │ │ ├── __init__.py │ │ ├── cmd_args_parser.py │ │ ├── dataset_setup.py │ │ ├── dataset_specific_params.py │ │ ├── hparams_parser.py │ │ ├── improved_cmd_args_parser.py │ │ ├── improved_cmd_args_parser_v2.py │ │ └── mixture_consistency.py ├── losses │ ├── __init__.py │ ├── norm.py │ ├── sisdr.py │ ├── snr.py │ └── test │ │ ├── test_normloss.py │ │ └── test_sisnr.py ├── models │ ├── __init__.py │ ├── attentive_sudormrf.py │ ├── attentive_sudormrf_v2.py │ ├── attentive_sudormrf_v3.py │ ├── causal_improved_sudormrf_v3.py │ ├── demucs.py │ ├── dprnn.py │ ├── groupcomm_sudormrf_v2.py │ ├── improved_sudormrf.py │ ├── original_convtasnet.py │ ├── sepformer.py │ ├── sudormrf.py │ └── two_step_tdcn.py └── utils │ ├── __init__.py │ ├── cometml_learned_masks.py │ ├── cometml_log_audio.py │ ├── cometml_loss_report.py │ ├── log_audio.py │ └── metrics_logger.py ├── notebooks ├── make_tables_and_plots.ipynb ├── sudormrf_extract_computation_metrics_example.ipynb ├── sudormrf_extract_sep_perf_metrics_example.ipynb └── sudormrf_how_to_use.ipynb └── utils ├── __init__.py ├── create_esc50_hierarchical.py ├── create_wsj0mix_hier_speakers.py ├── extract_model_performance.py ├── generate_data.sh ├── performance_results_parser.py ├── preprocess_wsj0mix.py ├── progress_display.py ├── run_performance_measures.sh └── simple_whamr_evaluation.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/README.md -------------------------------------------------------------------------------- /__config__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/__config__.py -------------------------------------------------------------------------------- /images/Selection_059.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/images/Selection_059.png -------------------------------------------------------------------------------- /images/Selection_060.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/images/Selection_060.png -------------------------------------------------------------------------------- /images/Selection_061.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/images/Selection_061.png -------------------------------------------------------------------------------- /images/Selection_062.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/images/Selection_062.png -------------------------------------------------------------------------------- /images/fuss_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/images/fuss_results.png -------------------------------------------------------------------------------- /pretrained_models/.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/pretrained_models/.gitattributes -------------------------------------------------------------------------------- /pretrained_models/download_pretrained_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/pretrained_models/download_pretrained_models.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/requirements.txt -------------------------------------------------------------------------------- /sudo_rm_rf/._.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/._.DS_Store -------------------------------------------------------------------------------- /sudo_rm_rf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/dataset_loader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/dataset_loader/abstract_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/dataset_loader/abstract_dataset.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/dataset_loader/augmented_mix_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/dataset_loader/augmented_mix_dataloader.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/dataset_loader/fuss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/dataset_loader/fuss.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/dataset_loader/libri2mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/dataset_loader/libri2mix.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/dataset_loader/musdb_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/dataset_loader/musdb_dataset.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/dataset_loader/torch_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/dataset_loader/torch_dataloader.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/dataset_loader/wham.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/dataset_loader/wham.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/dataset_loader/whamr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/dataset_loader/whamr.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/baseline_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/baseline_runner.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/run_attentive_sudormrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/run_attentive_sudormrf.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/run_fuss_separation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/run_fuss_separation.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/run_improved_sudormrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/run_improved_sudormrf.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/run_improved_sudormrf_reverb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/run_improved_sudormrf_reverb.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/run_sudormrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/run_sudormrf.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/run_sudormrf_gc_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/run_sudormrf_gc_v2.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/utils/cmd_args_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/utils/cmd_args_parser.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/utils/dataset_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/utils/dataset_setup.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/utils/dataset_specific_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/utils/dataset_specific_params.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/utils/hparams_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/utils/hparams_parser.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/utils/improved_cmd_args_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/utils/improved_cmd_args_parser.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/utils/improved_cmd_args_parser_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/utils/improved_cmd_args_parser_v2.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/experiments/utils/mixture_consistency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/experiments/utils/mixture_consistency.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/losses/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/losses/norm.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/losses/sisdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/losses/sisdr.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/losses/snr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/losses/snr.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/losses/test/test_normloss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/losses/test/test_normloss.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/losses/test/test_sisnr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/losses/test/test_sisnr.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/attentive_sudormrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/attentive_sudormrf.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/attentive_sudormrf_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/attentive_sudormrf_v2.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/attentive_sudormrf_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/attentive_sudormrf_v3.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/causal_improved_sudormrf_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/causal_improved_sudormrf_v3.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/demucs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/demucs.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/dprnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/dprnn.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/groupcomm_sudormrf_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/groupcomm_sudormrf_v2.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/improved_sudormrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/improved_sudormrf.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/original_convtasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/original_convtasnet.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/sepformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/sepformer.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/sudormrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/sudormrf.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/models/two_step_tdcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/models/two_step_tdcn.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/utils/cometml_learned_masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/utils/cometml_learned_masks.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/utils/cometml_log_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/utils/cometml_log_audio.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/utils/cometml_loss_report.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/utils/cometml_loss_report.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/utils/log_audio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/utils/log_audio.py -------------------------------------------------------------------------------- /sudo_rm_rf/dnn/utils/metrics_logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/dnn/utils/metrics_logger.py -------------------------------------------------------------------------------- /sudo_rm_rf/notebooks/make_tables_and_plots.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/notebooks/make_tables_and_plots.ipynb -------------------------------------------------------------------------------- /sudo_rm_rf/notebooks/sudormrf_extract_computation_metrics_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/notebooks/sudormrf_extract_computation_metrics_example.ipynb -------------------------------------------------------------------------------- /sudo_rm_rf/notebooks/sudormrf_extract_sep_perf_metrics_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/notebooks/sudormrf_extract_sep_perf_metrics_example.ipynb -------------------------------------------------------------------------------- /sudo_rm_rf/notebooks/sudormrf_how_to_use.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/notebooks/sudormrf_how_to_use.ipynb -------------------------------------------------------------------------------- /sudo_rm_rf/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /sudo_rm_rf/utils/create_esc50_hierarchical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/utils/create_esc50_hierarchical.py -------------------------------------------------------------------------------- /sudo_rm_rf/utils/create_wsj0mix_hier_speakers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/utils/create_wsj0mix_hier_speakers.py -------------------------------------------------------------------------------- /sudo_rm_rf/utils/extract_model_performance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/utils/extract_model_performance.py -------------------------------------------------------------------------------- /sudo_rm_rf/utils/generate_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/utils/generate_data.sh -------------------------------------------------------------------------------- /sudo_rm_rf/utils/performance_results_parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/utils/performance_results_parser.py -------------------------------------------------------------------------------- /sudo_rm_rf/utils/preprocess_wsj0mix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/utils/preprocess_wsj0mix.py -------------------------------------------------------------------------------- /sudo_rm_rf/utils/progress_display.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/utils/progress_display.py -------------------------------------------------------------------------------- /sudo_rm_rf/utils/run_performance_measures.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/utils/run_performance_measures.sh -------------------------------------------------------------------------------- /sudo_rm_rf/utils/simple_whamr_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/etzinis/sudo_rm_rf/HEAD/sudo_rm_rf/utils/simple_whamr_evaluation.py --------------------------------------------------------------------------------