├── .gitignore ├── README.md ├── datasets └── .gitignore ├── models └── model_no_augmentation.pt ├── report.pdf ├── src ├── data │ ├── adjacency.py │ ├── augmentation.py │ ├── calculate_distances.py │ ├── datasets.py │ ├── openpose_from_kth.sh │ ├── process_openpose.py │ └── util.py ├── lightning_model.py ├── network │ ├── st_graphconv.py │ └── stgcn.py ├── notebooks │ ├── .placeholder │ └── experiments.ipynb ├── scripts │ ├── KTH_train_model.py │ └── report │ │ ├── generate_figures.py │ │ ├── generate_log.py │ │ ├── img │ │ ├── batchnorm_layer2_off_acc.png │ │ ├── batchnorm_layer2_off_loss.png │ │ ├── dropout-0.25-train_loss.png │ │ ├── dropout-0.25-val_loss.png │ │ ├── dropout-0.5_lr_acc.png │ │ ├── dropout-0.5_lr_loss.png │ │ ├── unilabelling_res.png │ │ └── val_acc_data_aug.png │ │ └── logs │ │ ├── batchnorm_layer2_off_logs │ │ ├── train_acc.csv │ │ ├── train_loss.csv │ │ ├── val_acc.csv │ │ └── val_loss.csv │ │ ├── distance │ │ ├── run-version_3-tag-train_acc.csv │ │ ├── run-version_3-tag-train_loss.csv │ │ ├── run-version_3-tag-val_acc.csv │ │ └── run-version_3-tag-val_loss.csv │ │ ├── other │ │ ├── run-version_0-tag-val_acc.csv │ │ ├── run-version_2-tag-val_acc-2.csv │ │ ├── run-version_22-tag-val_acc.csv │ │ ├── run-version_22-tag-val_loss.csv │ │ ├── run-version_23-tag-val_acc.csv │ │ ├── run-version_23-tag-val_loss.csv │ │ ├── run-version_29-tag-train_loss.csv │ │ ├── run-version_29-tag-val_acc.csv │ │ ├── run-version_29-tag-val_loss.csv │ │ ├── run-version_30-tag-train_loss.csv │ │ ├── run-version_30-tag-val_acc.csv │ │ ├── run-version_31-tag-train_loss.csv │ │ ├── run-version_31-tag-val_acc.csv │ │ └── run-version_4-tag-val_acc.csv │ │ ├── spatial │ │ ├── run-version_4-tag-train_acc.csv │ │ ├── run-version_4-tag-train_loss.csv │ │ ├── run-version_4-tag-val_acc.csv │ │ └── run-version_4-tag-val_loss.csv │ │ ├── symmetrical │ │ ├── run-version_5-tag-train_acc.csv │ │ ├── run-version_5-tag-train_loss.csv │ │ ├── run-version_5-tag-val_acc.csv │ │ └── run-version_5-tag-val_loss.csv │ │ └── uni │ │ ├── run-version_1-tag-train_acc.csv │ │ ├── run-version_1-tag-train_loss.csv │ │ ├── run-version_1-tag-val_acc.csv │ │ └── run-version_1-tag-val_loss.csv └── utils │ ├── .placeholder │ └── model_utils.py └── stgcn.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/README.md -------------------------------------------------------------------------------- /datasets/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/datasets/.gitignore -------------------------------------------------------------------------------- /models/model_no_augmentation.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/models/model_no_augmentation.pt -------------------------------------------------------------------------------- /report.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/report.pdf -------------------------------------------------------------------------------- /src/data/adjacency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/data/adjacency.py -------------------------------------------------------------------------------- /src/data/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/data/augmentation.py -------------------------------------------------------------------------------- /src/data/calculate_distances.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/data/calculate_distances.py -------------------------------------------------------------------------------- /src/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/data/datasets.py -------------------------------------------------------------------------------- /src/data/openpose_from_kth.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/data/openpose_from_kth.sh -------------------------------------------------------------------------------- /src/data/process_openpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/data/process_openpose.py -------------------------------------------------------------------------------- /src/data/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/data/util.py -------------------------------------------------------------------------------- /src/lightning_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/lightning_model.py -------------------------------------------------------------------------------- /src/network/st_graphconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/network/st_graphconv.py -------------------------------------------------------------------------------- /src/network/stgcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/network/stgcn.py -------------------------------------------------------------------------------- /src/notebooks/.placeholder: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/notebooks/experiments.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/notebooks/experiments.ipynb -------------------------------------------------------------------------------- /src/scripts/KTH_train_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/KTH_train_model.py -------------------------------------------------------------------------------- /src/scripts/report/generate_figures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/generate_figures.py -------------------------------------------------------------------------------- /src/scripts/report/generate_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/generate_log.py -------------------------------------------------------------------------------- /src/scripts/report/img/batchnorm_layer2_off_acc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/img/batchnorm_layer2_off_acc.png -------------------------------------------------------------------------------- /src/scripts/report/img/batchnorm_layer2_off_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/img/batchnorm_layer2_off_loss.png -------------------------------------------------------------------------------- /src/scripts/report/img/dropout-0.25-train_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/img/dropout-0.25-train_loss.png -------------------------------------------------------------------------------- /src/scripts/report/img/dropout-0.25-val_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/img/dropout-0.25-val_loss.png -------------------------------------------------------------------------------- /src/scripts/report/img/dropout-0.5_lr_acc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/img/dropout-0.5_lr_acc.png -------------------------------------------------------------------------------- /src/scripts/report/img/dropout-0.5_lr_loss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/img/dropout-0.5_lr_loss.png -------------------------------------------------------------------------------- /src/scripts/report/img/unilabelling_res.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/img/unilabelling_res.png -------------------------------------------------------------------------------- /src/scripts/report/img/val_acc_data_aug.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/img/val_acc_data_aug.png -------------------------------------------------------------------------------- /src/scripts/report/logs/batchnorm_layer2_off_logs/train_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/batchnorm_layer2_off_logs/train_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/batchnorm_layer2_off_logs/train_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/batchnorm_layer2_off_logs/train_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/batchnorm_layer2_off_logs/val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/batchnorm_layer2_off_logs/val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/batchnorm_layer2_off_logs/val_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/batchnorm_layer2_off_logs/val_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/distance/run-version_3-tag-train_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/distance/run-version_3-tag-train_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/distance/run-version_3-tag-train_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/distance/run-version_3-tag-train_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/distance/run-version_3-tag-val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/distance/run-version_3-tag-val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/distance/run-version_3-tag-val_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/distance/run-version_3-tag-val_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_0-tag-val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_0-tag-val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_2-tag-val_acc-2.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_2-tag-val_acc-2.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_22-tag-val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_22-tag-val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_22-tag-val_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_22-tag-val_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_23-tag-val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_23-tag-val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_23-tag-val_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_23-tag-val_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_29-tag-train_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_29-tag-train_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_29-tag-val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_29-tag-val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_29-tag-val_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_29-tag-val_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_30-tag-train_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_30-tag-train_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_30-tag-val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_30-tag-val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_31-tag-train_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_31-tag-train_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_31-tag-val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_31-tag-val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/other/run-version_4-tag-val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/other/run-version_4-tag-val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/spatial/run-version_4-tag-train_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/spatial/run-version_4-tag-train_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/spatial/run-version_4-tag-train_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/spatial/run-version_4-tag-train_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/spatial/run-version_4-tag-val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/spatial/run-version_4-tag-val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/spatial/run-version_4-tag-val_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/spatial/run-version_4-tag-val_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/symmetrical/run-version_5-tag-train_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/symmetrical/run-version_5-tag-train_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/symmetrical/run-version_5-tag-train_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/symmetrical/run-version_5-tag-train_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/symmetrical/run-version_5-tag-val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/symmetrical/run-version_5-tag-val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/symmetrical/run-version_5-tag-val_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/symmetrical/run-version_5-tag-val_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/uni/run-version_1-tag-train_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/uni/run-version_1-tag-train_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/uni/run-version_1-tag-train_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/uni/run-version_1-tag-train_loss.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/uni/run-version_1-tag-val_acc.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/uni/run-version_1-tag-val_acc.csv -------------------------------------------------------------------------------- /src/scripts/report/logs/uni/run-version_1-tag-val_loss.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/scripts/report/logs/uni/run-version_1-tag-val_loss.csv -------------------------------------------------------------------------------- /src/utils/.placeholder: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/src/utils/model_utils.py -------------------------------------------------------------------------------- /stgcn.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nagyrajmund/st-gcn/HEAD/stgcn.txt --------------------------------------------------------------------------------