├── LICENSE ├── README.md ├── environment.yml ├── figures └── outline_simple_2.png ├── jq └── jq-win64.exe ├── requirments.txt └── src ├── __init__.py ├── batch_train.sh ├── config.json ├── logger_config.json ├── models ├── __init__.py ├── main_painAttnNet.py ├── module_mscn.py ├── module_se_resnet.py └── module_transformer_encoder.py ├── parser.py ├── tests ├── __init__.py ├── test_PainAttnNet_output.py ├── test_generate_kfolds_index.py ├── test_mscn.py ├── test_process_bioVid.py ├── test_se_resnet.py └── test_transformer_encoder.py ├── train_kfold_cv.py ├── trainers ├── __init__.py ├── checkpoint_handler.py ├── device_prep.py ├── main_trainer.py └── metrics_manager.py └── utils ├── __init__.py ├── process_bioVid.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/README.md -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/environment.yml -------------------------------------------------------------------------------- /figures/outline_simple_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/figures/outline_simple_2.png -------------------------------------------------------------------------------- /jq/jq-win64.exe: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/jq/jq-win64.exe -------------------------------------------------------------------------------- /requirments.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/requirments.txt -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/batch_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/batch_train.sh -------------------------------------------------------------------------------- /src/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/config.json -------------------------------------------------------------------------------- /src/logger_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/logger_config.json -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/main_painAttnNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/models/main_painAttnNet.py -------------------------------------------------------------------------------- /src/models/module_mscn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/models/module_mscn.py -------------------------------------------------------------------------------- /src/models/module_se_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/models/module_se_resnet.py -------------------------------------------------------------------------------- /src/models/module_transformer_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/models/module_transformer_encoder.py -------------------------------------------------------------------------------- /src/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/parser.py -------------------------------------------------------------------------------- /src/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/tests/test_PainAttnNet_output.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/tests/test_PainAttnNet_output.py -------------------------------------------------------------------------------- /src/tests/test_generate_kfolds_index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/tests/test_generate_kfolds_index.py -------------------------------------------------------------------------------- /src/tests/test_mscn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/tests/test_mscn.py -------------------------------------------------------------------------------- /src/tests/test_process_bioVid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/tests/test_process_bioVid.py -------------------------------------------------------------------------------- /src/tests/test_se_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/tests/test_se_resnet.py -------------------------------------------------------------------------------- /src/tests/test_transformer_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/tests/test_transformer_encoder.py -------------------------------------------------------------------------------- /src/train_kfold_cv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/train_kfold_cv.py -------------------------------------------------------------------------------- /src/trainers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/trainers/checkpoint_handler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/trainers/checkpoint_handler.py -------------------------------------------------------------------------------- /src/trainers/device_prep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/trainers/device_prep.py -------------------------------------------------------------------------------- /src/trainers/main_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/trainers/main_trainer.py -------------------------------------------------------------------------------- /src/trainers/metrics_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/trainers/metrics_manager.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/process_bioVid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/utils/process_bioVid.py -------------------------------------------------------------------------------- /src/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhenyuanlu/PainAttnNet/HEAD/src/utils/utils.py --------------------------------------------------------------------------------