├── README.md ├── loader ├── data_clean.py ├── data_loader1.py ├── data_output.py ├── database.py ├── ni_database.py └── ni_loader.py ├── mainer ├── evaluate.py ├── evaluate_output.py ├── main_TEST.py ├── main_att.py ├── main_baseline.py ├── main_compare.py ├── main_featurefusion.py ├── main_model516.py ├── main_model519.py ├── main_model520.py ├── main_ni_ada.py ├── main_ni_baseline_ada.py ├── main_ni_compare.py ├── main_tranlstm.py ├── main_transformer.py └── train_test.py ├── model ├── base_transformer │ ├── custom │ │ ├── custom_activation.py │ │ ├── custom_opt.py │ │ ├── dilate_loss.py │ │ └── focal_loss.py │ ├── meta_models │ │ ├── basic_ae.py │ │ └── merging_model.py │ ├── trainer.py │ ├── transformer_basic.py │ └── transformer_xl │ │ ├── attn.py │ │ ├── data_embedding.py │ │ ├── dummy_torch.py │ │ ├── informer.py │ │ ├── lower_upper_config.py │ │ ├── masks.py │ │ ├── multi_head_base.py │ │ ├── transformer_basic.py │ │ ├── transformer_bottleneck.py │ │ └── transformer_xl.py ├── baseline │ ├── baseline.py │ ├── params.py │ ├── params_ni.py │ └── trainer.py ├── featurefusion │ ├── criterions.py │ ├── params.json │ ├── params.py │ ├── params_save.py │ ├── readme │ ├── trainer.py │ └── tranlstm.py └── tranlstm │ ├── criterions.py │ ├── params.json │ ├── params.py │ ├── params_save.py │ ├── readme │ ├── trainer.py │ └── tranlstm.py └── picture ├── 3_method_compare.png └── net.jpg /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/README.md -------------------------------------------------------------------------------- /loader/data_clean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/loader/data_clean.py -------------------------------------------------------------------------------- /loader/data_loader1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/loader/data_loader1.py -------------------------------------------------------------------------------- /loader/data_output.py: -------------------------------------------------------------------------------- 1 | 2 | 3 | -------------------------------------------------------------------------------- /loader/database.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/loader/database.py -------------------------------------------------------------------------------- /loader/ni_database.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/loader/ni_database.py -------------------------------------------------------------------------------- /loader/ni_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/loader/ni_loader.py -------------------------------------------------------------------------------- /mainer/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/evaluate.py -------------------------------------------------------------------------------- /mainer/evaluate_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/evaluate_output.py -------------------------------------------------------------------------------- /mainer/main_TEST.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_TEST.py -------------------------------------------------------------------------------- /mainer/main_att.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_att.py -------------------------------------------------------------------------------- /mainer/main_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_baseline.py -------------------------------------------------------------------------------- /mainer/main_compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_compare.py -------------------------------------------------------------------------------- /mainer/main_featurefusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_featurefusion.py -------------------------------------------------------------------------------- /mainer/main_model516.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_model516.py -------------------------------------------------------------------------------- /mainer/main_model519.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_model519.py -------------------------------------------------------------------------------- /mainer/main_model520.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_model520.py -------------------------------------------------------------------------------- /mainer/main_ni_ada.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_ni_ada.py -------------------------------------------------------------------------------- /mainer/main_ni_baseline_ada.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_ni_baseline_ada.py -------------------------------------------------------------------------------- /mainer/main_ni_compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_ni_compare.py -------------------------------------------------------------------------------- /mainer/main_tranlstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_tranlstm.py -------------------------------------------------------------------------------- /mainer/main_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/main_transformer.py -------------------------------------------------------------------------------- /mainer/train_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/mainer/train_test.py -------------------------------------------------------------------------------- /model/base_transformer/custom/custom_activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/custom/custom_activation.py -------------------------------------------------------------------------------- /model/base_transformer/custom/custom_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/custom/custom_opt.py -------------------------------------------------------------------------------- /model/base_transformer/custom/dilate_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/custom/dilate_loss.py -------------------------------------------------------------------------------- /model/base_transformer/custom/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/custom/focal_loss.py -------------------------------------------------------------------------------- /model/base_transformer/meta_models/basic_ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/meta_models/basic_ae.py -------------------------------------------------------------------------------- /model/base_transformer/meta_models/merging_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/meta_models/merging_model.py -------------------------------------------------------------------------------- /model/base_transformer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/trainer.py -------------------------------------------------------------------------------- /model/base_transformer/transformer_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/transformer_basic.py -------------------------------------------------------------------------------- /model/base_transformer/transformer_xl/attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/transformer_xl/attn.py -------------------------------------------------------------------------------- /model/base_transformer/transformer_xl/data_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/transformer_xl/data_embedding.py -------------------------------------------------------------------------------- /model/base_transformer/transformer_xl/dummy_torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/transformer_xl/dummy_torch.py -------------------------------------------------------------------------------- /model/base_transformer/transformer_xl/informer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/transformer_xl/informer.py -------------------------------------------------------------------------------- /model/base_transformer/transformer_xl/lower_upper_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/transformer_xl/lower_upper_config.py -------------------------------------------------------------------------------- /model/base_transformer/transformer_xl/masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/transformer_xl/masks.py -------------------------------------------------------------------------------- /model/base_transformer/transformer_xl/multi_head_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/transformer_xl/multi_head_base.py -------------------------------------------------------------------------------- /model/base_transformer/transformer_xl/transformer_basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/transformer_xl/transformer_basic.py -------------------------------------------------------------------------------- /model/base_transformer/transformer_xl/transformer_bottleneck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/transformer_xl/transformer_bottleneck.py -------------------------------------------------------------------------------- /model/base_transformer/transformer_xl/transformer_xl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/base_transformer/transformer_xl/transformer_xl.py -------------------------------------------------------------------------------- /model/baseline/baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/baseline/baseline.py -------------------------------------------------------------------------------- /model/baseline/params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/baseline/params.py -------------------------------------------------------------------------------- /model/baseline/params_ni.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/baseline/params_ni.py -------------------------------------------------------------------------------- /model/baseline/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/baseline/trainer.py -------------------------------------------------------------------------------- /model/featurefusion/criterions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/featurefusion/criterions.py -------------------------------------------------------------------------------- /model/featurefusion/params.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/featurefusion/params.json -------------------------------------------------------------------------------- /model/featurefusion/params.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/featurefusion/params_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/featurefusion/params_save.py -------------------------------------------------------------------------------- /model/featurefusion/readme: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/featurefusion/readme -------------------------------------------------------------------------------- /model/featurefusion/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/featurefusion/trainer.py -------------------------------------------------------------------------------- /model/featurefusion/tranlstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/featurefusion/tranlstm.py -------------------------------------------------------------------------------- /model/tranlstm/criterions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/tranlstm/criterions.py -------------------------------------------------------------------------------- /model/tranlstm/params.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/tranlstm/params.json -------------------------------------------------------------------------------- /model/tranlstm/params.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/tranlstm/params_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/tranlstm/params_save.py -------------------------------------------------------------------------------- /model/tranlstm/readme: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/tranlstm/readme -------------------------------------------------------------------------------- /model/tranlstm/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/tranlstm/trainer.py -------------------------------------------------------------------------------- /model/tranlstm/tranlstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/model/tranlstm/tranlstm.py -------------------------------------------------------------------------------- /picture/3_method_compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/picture/3_method_compare.png -------------------------------------------------------------------------------- /picture/net.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/heeeyk/Transformer-DOA-Prediction/HEAD/picture/net.jpg --------------------------------------------------------------------------------