├── .gitignore ├── Makefile ├── analyze_complexity.py ├── analyze_simulated.py ├── checkpoints ├── cross-3d-directional.bin ├── cross-3d-omni.bin ├── doanet.bin ├── neural-srp-directional.bin ├── neural-srp-multi.bin ├── neural-srp-omni.bin └── neural-srp-rand.bin ├── datasets ├── __init__.py ├── array_setup.py ├── hnet_dataset_generator.py ├── librispeech_dataset.py ├── locata_dataset.py ├── mic_pos_utils.py ├── preprocess_tau_nigens_dataset.py ├── random_trajectory_dataset.py └── tau_nigens_dataset.py ├── example.png ├── hnet_model.h5 ├── loss.py ├── metrics.py ├── models ├── __init__.py ├── cross_3d.py ├── doanet.py ├── hnet.py ├── layers.py ├── mic_selection.py ├── neural_srp.py ├── numpy_transforms.py ├── nw_gcc.py ├── signal_processing.py └── srp.py ├── params.json ├── qsub.pbs ├── readme.md ├── test_simulated.py ├── tests ├── __init__.py ├── test_array_setup.py ├── test_mic_pos_utils.py ├── test_mic_selection.py ├── test_signal_processing.py └── test_utils.py ├── train_multisource.py ├── train_singlesource.py ├── trainers ├── cross_3d.py ├── hnet.py ├── neural_srp_one_source.py ├── one_source_tracker.py └── tau_nigens.py ├── utils.py ├── visualize_locata.py ├── visualize_tau.py └── viz_snr_rt60.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/.gitignore -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/Makefile -------------------------------------------------------------------------------- /analyze_complexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/analyze_complexity.py -------------------------------------------------------------------------------- /analyze_simulated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/analyze_simulated.py -------------------------------------------------------------------------------- /checkpoints/cross-3d-directional.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/checkpoints/cross-3d-directional.bin -------------------------------------------------------------------------------- /checkpoints/cross-3d-omni.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/checkpoints/cross-3d-omni.bin -------------------------------------------------------------------------------- /checkpoints/doanet.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/checkpoints/doanet.bin -------------------------------------------------------------------------------- /checkpoints/neural-srp-directional.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/checkpoints/neural-srp-directional.bin -------------------------------------------------------------------------------- /checkpoints/neural-srp-multi.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/checkpoints/neural-srp-multi.bin -------------------------------------------------------------------------------- /checkpoints/neural-srp-omni.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/checkpoints/neural-srp-omni.bin -------------------------------------------------------------------------------- /checkpoints/neural-srp-rand.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/checkpoints/neural-srp-rand.bin -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/array_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/datasets/array_setup.py -------------------------------------------------------------------------------- /datasets/hnet_dataset_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/datasets/hnet_dataset_generator.py -------------------------------------------------------------------------------- /datasets/librispeech_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/datasets/librispeech_dataset.py -------------------------------------------------------------------------------- /datasets/locata_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/datasets/locata_dataset.py -------------------------------------------------------------------------------- /datasets/mic_pos_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/datasets/mic_pos_utils.py -------------------------------------------------------------------------------- /datasets/preprocess_tau_nigens_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/datasets/preprocess_tau_nigens_dataset.py -------------------------------------------------------------------------------- /datasets/random_trajectory_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/datasets/random_trajectory_dataset.py -------------------------------------------------------------------------------- /datasets/tau_nigens_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/datasets/tau_nigens_dataset.py -------------------------------------------------------------------------------- /example.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/example.png -------------------------------------------------------------------------------- /hnet_model.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/hnet_model.h5 -------------------------------------------------------------------------------- /loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/loss.py -------------------------------------------------------------------------------- /metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/metrics.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/cross_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/models/cross_3d.py -------------------------------------------------------------------------------- /models/doanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/models/doanet.py -------------------------------------------------------------------------------- /models/hnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/models/hnet.py -------------------------------------------------------------------------------- /models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/models/layers.py -------------------------------------------------------------------------------- /models/mic_selection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/models/mic_selection.py -------------------------------------------------------------------------------- /models/neural_srp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/models/neural_srp.py -------------------------------------------------------------------------------- /models/numpy_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/models/numpy_transforms.py -------------------------------------------------------------------------------- /models/nw_gcc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/models/nw_gcc.py -------------------------------------------------------------------------------- /models/signal_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/models/signal_processing.py -------------------------------------------------------------------------------- /models/srp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/models/srp.py -------------------------------------------------------------------------------- /params.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/params.json -------------------------------------------------------------------------------- /qsub.pbs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/qsub.pbs -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/readme.md -------------------------------------------------------------------------------- /test_simulated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/test_simulated.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_array_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/tests/test_array_setup.py -------------------------------------------------------------------------------- /tests/test_mic_pos_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/tests/test_mic_pos_utils.py -------------------------------------------------------------------------------- /tests/test_mic_selection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/tests/test_mic_selection.py -------------------------------------------------------------------------------- /tests/test_signal_processing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/tests/test_signal_processing.py -------------------------------------------------------------------------------- /tests/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/tests/test_utils.py -------------------------------------------------------------------------------- /train_multisource.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/train_multisource.py -------------------------------------------------------------------------------- /train_singlesource.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/train_singlesource.py -------------------------------------------------------------------------------- /trainers/cross_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/trainers/cross_3d.py -------------------------------------------------------------------------------- /trainers/hnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/trainers/hnet.py -------------------------------------------------------------------------------- /trainers/neural_srp_one_source.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/trainers/neural_srp_one_source.py -------------------------------------------------------------------------------- /trainers/one_source_tracker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/trainers/one_source_tracker.py -------------------------------------------------------------------------------- /trainers/tau_nigens.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/trainers/tau_nigens.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/utils.py -------------------------------------------------------------------------------- /visualize_locata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/visualize_locata.py -------------------------------------------------------------------------------- /visualize_tau.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/visualize_tau.py -------------------------------------------------------------------------------- /viz_snr_rt60.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/egrinstein/neural_srp/HEAD/viz_snr_rt60.py --------------------------------------------------------------------------------