├── .gitignore ├── LICENSE ├── README.md ├── Speaker_Embed └── extract_embedding.py ├── config ├── finetune_English.json ├── finetune_Mandarain.json └── pretrain.json ├── convert_AMI2Libri.py ├── convert_ASR2aishell1.py ├── datasets ├── AMI.json ├── ASR.json ├── dataset_loader.py ├── dataset_util.py └── generate_dataset.py ├── eval_conversation.py ├── image ├── arch.png └── cover.png ├── plot_result.py └── src ├── .DS_Store ├── __init__.py ├── datasets ├── dataset_AMI_Libri.py └── dataset_TSE.py ├── hl_modules ├── conversation_based_hl_module.py └── distance_based_hl_module.py ├── losses ├── .DS_Store ├── LogPowerLoss.py ├── MultiResoLoss.py ├── Perceptual_Loss.py ├── SNRLP.py ├── SNRLosses.py ├── mask.py └── stft.py ├── metrics └── metrics.py ├── models ├── Conversation_sep │ └── net.py └── Sep_module │ └── separator.py ├── train.py ├── training └── tain_val.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/README.md -------------------------------------------------------------------------------- /Speaker_Embed/extract_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/Speaker_Embed/extract_embedding.py -------------------------------------------------------------------------------- /config/finetune_English.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/config/finetune_English.json -------------------------------------------------------------------------------- /config/finetune_Mandarain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/config/finetune_Mandarain.json -------------------------------------------------------------------------------- /config/pretrain.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/config/pretrain.json -------------------------------------------------------------------------------- /convert_AMI2Libri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/convert_AMI2Libri.py -------------------------------------------------------------------------------- /convert_ASR2aishell1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/convert_ASR2aishell1.py -------------------------------------------------------------------------------- /datasets/AMI.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/datasets/AMI.json -------------------------------------------------------------------------------- /datasets/ASR.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/datasets/ASR.json -------------------------------------------------------------------------------- /datasets/dataset_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/datasets/dataset_loader.py -------------------------------------------------------------------------------- /datasets/dataset_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/datasets/dataset_util.py -------------------------------------------------------------------------------- /datasets/generate_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/datasets/generate_dataset.py -------------------------------------------------------------------------------- /eval_conversation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/eval_conversation.py -------------------------------------------------------------------------------- /image/arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/image/arch.png -------------------------------------------------------------------------------- /image/cover.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/image/cover.png -------------------------------------------------------------------------------- /plot_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/plot_result.py -------------------------------------------------------------------------------- /src/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/.DS_Store -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/datasets/dataset_AMI_Libri.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/datasets/dataset_AMI_Libri.py -------------------------------------------------------------------------------- /src/datasets/dataset_TSE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/datasets/dataset_TSE.py -------------------------------------------------------------------------------- /src/hl_modules/conversation_based_hl_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/hl_modules/conversation_based_hl_module.py -------------------------------------------------------------------------------- /src/hl_modules/distance_based_hl_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/hl_modules/distance_based_hl_module.py -------------------------------------------------------------------------------- /src/losses/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/losses/.DS_Store -------------------------------------------------------------------------------- /src/losses/LogPowerLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/losses/LogPowerLoss.py -------------------------------------------------------------------------------- /src/losses/MultiResoLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/losses/MultiResoLoss.py -------------------------------------------------------------------------------- /src/losses/Perceptual_Loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/losses/Perceptual_Loss.py -------------------------------------------------------------------------------- /src/losses/SNRLP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/losses/SNRLP.py -------------------------------------------------------------------------------- /src/losses/SNRLosses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/losses/SNRLosses.py -------------------------------------------------------------------------------- /src/losses/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/losses/mask.py -------------------------------------------------------------------------------- /src/losses/stft.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/losses/stft.py -------------------------------------------------------------------------------- /src/metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/metrics/metrics.py -------------------------------------------------------------------------------- /src/models/Conversation_sep/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/models/Conversation_sep/net.py -------------------------------------------------------------------------------- /src/models/Sep_module/separator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/models/Sep_module/separator.py -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/train.py -------------------------------------------------------------------------------- /src/training/tain_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/training/tain_val.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chentuochao/Target-Conversation-Extraction/HEAD/src/utils.py --------------------------------------------------------------------------------