├── .gitignore ├── LICENSE ├── README.md ├── configs ├── embed.json └── tsh.json ├── data └── .keep ├── requirements.txt ├── slurm.py └── src ├── binaural_embed_pl_module.py ├── datasets ├── MixLibriSpeech.py ├── MixLibriSpeechDM.py ├── MixLibriSpeechMotion.py ├── MixLibriSpeechNoisyEnroll.py ├── MixLibriSpeechNoisyEnrollDirectional.py ├── MixLibriSpeechNoisyEnrollNorm.py ├── MixLibriSpeechNoisyEnrollWP.py ├── OracleMixLibriSpeech.py ├── SpeechSeparationDataset.py ├── augmentations.py ├── dvector_embeddings.py ├── generate_jams.py ├── librispeech2scaper.py ├── motion_simulator.py ├── multi_ch_simulator.py └── nemo_embed.py ├── eval └── binaural.py ├── losses ├── LossFn.py ├── fused_loss.py ├── perceptual_losses.py ├── scale_dependent_snr_loss.py ├── sisdr_with_pit.py └── sisdr_with_sum_loss.py ├── models ├── tfgridnet_orig │ ├── stft.py │ ├── stft_decoder.py │ └── tfgridnet.py └── tfgridnet_realtime │ ├── film.py │ ├── net.py │ └── tfgridnet_causal.py ├── trainer.py ├── ts_hear_embed_pl_module.py ├── ts_hear_test.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/README.md -------------------------------------------------------------------------------- /configs/embed.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/configs/embed.json -------------------------------------------------------------------------------- /configs/tsh.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/configs/tsh.json -------------------------------------------------------------------------------- /data/.keep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/requirements.txt -------------------------------------------------------------------------------- /slurm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/slurm.py -------------------------------------------------------------------------------- /src/binaural_embed_pl_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/binaural_embed_pl_module.py -------------------------------------------------------------------------------- /src/datasets/MixLibriSpeech.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/MixLibriSpeech.py -------------------------------------------------------------------------------- /src/datasets/MixLibriSpeechDM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/MixLibriSpeechDM.py -------------------------------------------------------------------------------- /src/datasets/MixLibriSpeechMotion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/MixLibriSpeechMotion.py -------------------------------------------------------------------------------- /src/datasets/MixLibriSpeechNoisyEnroll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/MixLibriSpeechNoisyEnroll.py -------------------------------------------------------------------------------- /src/datasets/MixLibriSpeechNoisyEnrollDirectional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/MixLibriSpeechNoisyEnrollDirectional.py -------------------------------------------------------------------------------- /src/datasets/MixLibriSpeechNoisyEnrollNorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/MixLibriSpeechNoisyEnrollNorm.py -------------------------------------------------------------------------------- /src/datasets/MixLibriSpeechNoisyEnrollWP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/MixLibriSpeechNoisyEnrollWP.py -------------------------------------------------------------------------------- /src/datasets/OracleMixLibriSpeech.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/OracleMixLibriSpeech.py -------------------------------------------------------------------------------- /src/datasets/SpeechSeparationDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/SpeechSeparationDataset.py -------------------------------------------------------------------------------- /src/datasets/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/augmentations.py -------------------------------------------------------------------------------- /src/datasets/dvector_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/dvector_embeddings.py -------------------------------------------------------------------------------- /src/datasets/generate_jams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/generate_jams.py -------------------------------------------------------------------------------- /src/datasets/librispeech2scaper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/librispeech2scaper.py -------------------------------------------------------------------------------- /src/datasets/motion_simulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/motion_simulator.py -------------------------------------------------------------------------------- /src/datasets/multi_ch_simulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/multi_ch_simulator.py -------------------------------------------------------------------------------- /src/datasets/nemo_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/datasets/nemo_embed.py -------------------------------------------------------------------------------- /src/eval/binaural.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/eval/binaural.py -------------------------------------------------------------------------------- /src/losses/LossFn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/losses/LossFn.py -------------------------------------------------------------------------------- /src/losses/fused_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/losses/fused_loss.py -------------------------------------------------------------------------------- /src/losses/perceptual_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/losses/perceptual_losses.py -------------------------------------------------------------------------------- /src/losses/scale_dependent_snr_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/losses/scale_dependent_snr_loss.py -------------------------------------------------------------------------------- /src/losses/sisdr_with_pit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/losses/sisdr_with_pit.py -------------------------------------------------------------------------------- /src/losses/sisdr_with_sum_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/losses/sisdr_with_sum_loss.py -------------------------------------------------------------------------------- /src/models/tfgridnet_orig/stft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/models/tfgridnet_orig/stft.py -------------------------------------------------------------------------------- /src/models/tfgridnet_orig/stft_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/models/tfgridnet_orig/stft_decoder.py -------------------------------------------------------------------------------- /src/models/tfgridnet_orig/tfgridnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/models/tfgridnet_orig/tfgridnet.py -------------------------------------------------------------------------------- /src/models/tfgridnet_realtime/film.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/models/tfgridnet_realtime/film.py -------------------------------------------------------------------------------- /src/models/tfgridnet_realtime/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/models/tfgridnet_realtime/net.py -------------------------------------------------------------------------------- /src/models/tfgridnet_realtime/tfgridnet_causal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/models/tfgridnet_realtime/tfgridnet_causal.py -------------------------------------------------------------------------------- /src/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/trainer.py -------------------------------------------------------------------------------- /src/ts_hear_embed_pl_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/ts_hear_embed_pl_module.py -------------------------------------------------------------------------------- /src/ts_hear_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/ts_hear_test.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vb000/LookOnceToHear/HEAD/src/utils.py --------------------------------------------------------------------------------