├── .gitignore ├── LICENSE ├── README.md ├── data ├── 2speakers ├── cv │ ├── mix.scp │ ├── s1.scp │ └── s2.scp ├── tr │ ├── mix.scp │ ├── s1.scp │ └── s2.scp └── tt │ ├── mix.scp │ ├── s1.scp │ └── s2.scp ├── data_test ├── cv │ ├── mix.scp │ ├── s1.scp │ └── s2.scp ├── tr │ ├── mix.scp │ ├── s1.scp │ └── s2.scp └── tt │ ├── mix.scp │ ├── s1.scp │ └── s2.scp ├── model ├── __init__.py ├── layer_norm.py ├── misc.py ├── modules.py ├── show.py └── tasnet.py ├── requirements.txt ├── scripts ├── debug.sh ├── prep_egs.sh ├── run_oracle_mask.sh ├── run_tasnet.sh ├── unzip.sh └── wav_duration.sh ├── steps ├── iam_oracle.py ├── iam_oracle_phase.py ├── iam_oracle_statistic.py ├── iam_oracle_statistic_log.py ├── iam_oracle_time_domain.py ├── ibm_oracle.py ├── ibm_oracle_phase.py ├── ibm_oracle_time_domain.py ├── ipsm_oracle.py ├── ipsm_oracle_phase.py ├── ipsm_oracle_statistic.py ├── ipsm_oracle_statistic_log.py ├── irm_oracle.py ├── irm_oracle_phase.py ├── irm_oracle_time_domain.py └── run_tasnet.py └── utils ├── base ├── __init__.py ├── data_reader.py ├── data_reader.py.bak ├── dataset.py └── misc.py ├── evaluate ├── est_list_prep.py ├── eval_sdr.py ├── eval_sdr_sources.py ├── eval_si_sdr.py ├── si_sdr_numpy.py ├── si_sdr_torch.py └── spk2gender ├── misc ├── common.py └── logging.py └── sigproc ├── dsp.py ├── interpolate_and_smooth.py ├── mask.py ├── sigproc.py ├── spectrum.py └── time_domain_mask.py /.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__/ 2 | exp/ 3 | *.log 4 | data/2speakers 5 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /data/2speakers: -------------------------------------------------------------------------------- 1 | /mnt/data/2speakers -------------------------------------------------------------------------------- /data/cv/mix.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data/cv/mix.scp -------------------------------------------------------------------------------- /data/cv/s1.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data/cv/s1.scp -------------------------------------------------------------------------------- /data/cv/s2.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data/cv/s2.scp -------------------------------------------------------------------------------- /data/tr/mix.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data/tr/mix.scp -------------------------------------------------------------------------------- /data/tr/s1.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data/tr/s1.scp -------------------------------------------------------------------------------- /data/tr/s2.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data/tr/s2.scp -------------------------------------------------------------------------------- /data/tt/mix.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data/tt/mix.scp -------------------------------------------------------------------------------- /data/tt/s1.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data/tt/s1.scp -------------------------------------------------------------------------------- /data/tt/s2.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data/tt/s2.scp -------------------------------------------------------------------------------- /data_test/cv/mix.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data_test/cv/mix.scp -------------------------------------------------------------------------------- /data_test/cv/s1.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data_test/cv/s1.scp -------------------------------------------------------------------------------- /data_test/cv/s2.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data_test/cv/s2.scp -------------------------------------------------------------------------------- /data_test/tr/mix.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data_test/tr/mix.scp -------------------------------------------------------------------------------- /data_test/tr/s1.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data_test/tr/s1.scp -------------------------------------------------------------------------------- /data_test/tr/s2.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data_test/tr/s2.scp -------------------------------------------------------------------------------- /data_test/tt/mix.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data_test/tt/mix.scp -------------------------------------------------------------------------------- /data_test/tt/s1.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data_test/tt/s1.scp -------------------------------------------------------------------------------- /data_test/tt/s2.scp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/data_test/tt/s2.scp -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/model/layer_norm.py -------------------------------------------------------------------------------- /model/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/model/misc.py -------------------------------------------------------------------------------- /model/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/model/modules.py -------------------------------------------------------------------------------- /model/show.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/model/show.py -------------------------------------------------------------------------------- /model/tasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/model/tasnet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | tensorboardX 2 | soundfile 3 | librosa 4 | museval 5 | -------------------------------------------------------------------------------- /scripts/debug.sh: -------------------------------------------------------------------------------- 1 | nvidia-smi -------------------------------------------------------------------------------- /scripts/prep_egs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/scripts/prep_egs.sh -------------------------------------------------------------------------------- /scripts/run_oracle_mask.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/scripts/run_oracle_mask.sh -------------------------------------------------------------------------------- /scripts/run_tasnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/scripts/run_tasnet.sh -------------------------------------------------------------------------------- /scripts/unzip.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/scripts/unzip.sh -------------------------------------------------------------------------------- /scripts/wav_duration.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/scripts/wav_duration.sh -------------------------------------------------------------------------------- /steps/iam_oracle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/iam_oracle.py -------------------------------------------------------------------------------- /steps/iam_oracle_phase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/iam_oracle_phase.py -------------------------------------------------------------------------------- /steps/iam_oracle_statistic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/iam_oracle_statistic.py -------------------------------------------------------------------------------- /steps/iam_oracle_statistic_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/iam_oracle_statistic_log.py -------------------------------------------------------------------------------- /steps/iam_oracle_time_domain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/iam_oracle_time_domain.py -------------------------------------------------------------------------------- /steps/ibm_oracle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/ibm_oracle.py -------------------------------------------------------------------------------- /steps/ibm_oracle_phase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/ibm_oracle_phase.py -------------------------------------------------------------------------------- /steps/ibm_oracle_time_domain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/ibm_oracle_time_domain.py -------------------------------------------------------------------------------- /steps/ipsm_oracle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/ipsm_oracle.py -------------------------------------------------------------------------------- /steps/ipsm_oracle_phase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/ipsm_oracle_phase.py -------------------------------------------------------------------------------- /steps/ipsm_oracle_statistic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/ipsm_oracle_statistic.py -------------------------------------------------------------------------------- /steps/ipsm_oracle_statistic_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/ipsm_oracle_statistic_log.py -------------------------------------------------------------------------------- /steps/irm_oracle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/irm_oracle.py -------------------------------------------------------------------------------- /steps/irm_oracle_phase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/irm_oracle_phase.py -------------------------------------------------------------------------------- /steps/irm_oracle_time_domain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/irm_oracle_time_domain.py -------------------------------------------------------------------------------- /steps/run_tasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/steps/run_tasnet.py -------------------------------------------------------------------------------- /utils/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/base/__init__.py -------------------------------------------------------------------------------- /utils/base/data_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/base/data_reader.py -------------------------------------------------------------------------------- /utils/base/data_reader.py.bak: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/base/data_reader.py.bak -------------------------------------------------------------------------------- /utils/base/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/base/dataset.py -------------------------------------------------------------------------------- /utils/base/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/base/misc.py -------------------------------------------------------------------------------- /utils/evaluate/est_list_prep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/evaluate/est_list_prep.py -------------------------------------------------------------------------------- /utils/evaluate/eval_sdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/evaluate/eval_sdr.py -------------------------------------------------------------------------------- /utils/evaluate/eval_sdr_sources.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/evaluate/eval_sdr_sources.py -------------------------------------------------------------------------------- /utils/evaluate/eval_si_sdr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/evaluate/eval_si_sdr.py -------------------------------------------------------------------------------- /utils/evaluate/si_sdr_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/evaluate/si_sdr_numpy.py -------------------------------------------------------------------------------- /utils/evaluate/si_sdr_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/evaluate/si_sdr_torch.py -------------------------------------------------------------------------------- /utils/evaluate/spk2gender: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/evaluate/spk2gender -------------------------------------------------------------------------------- /utils/misc/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/misc/common.py -------------------------------------------------------------------------------- /utils/misc/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/misc/logging.py -------------------------------------------------------------------------------- /utils/sigproc/dsp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/sigproc/dsp.py -------------------------------------------------------------------------------- /utils/sigproc/interpolate_and_smooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/sigproc/interpolate_and_smooth.py -------------------------------------------------------------------------------- /utils/sigproc/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/sigproc/mask.py -------------------------------------------------------------------------------- /utils/sigproc/sigproc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/sigproc/sigproc.py -------------------------------------------------------------------------------- /utils/sigproc/spectrum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/sigproc/spectrum.py -------------------------------------------------------------------------------- /utils/sigproc/time_domain_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wangkenpu/Conv-TasNet-PyTorch/HEAD/utils/sigproc/time_domain_mask.py --------------------------------------------------------------------------------