├── .gitignore ├── LICENSE ├── README.md ├── config ├── uav-cross-subjectv1 │ ├── test.yaml │ ├── train.yaml │ └── train_shiftgcn.yaml └── uav-cross-subjectv2 │ ├── test.yaml │ ├── train.yaml │ └── train_shiftgcn.yaml ├── data └── data_here ├── data_gen ├── __init__.py ├── gen_bone_data.py ├── gen_motion_data.py ├── kinetics_gendata.py ├── merge_joint_bone_data.py ├── ntu_gendata.py ├── preprocess.py ├── rotation.py └── uavhumanposetools │ ├── README.md │ ├── generate_data.py │ ├── pose_data_tools │ ├── graph.py │ ├── preprocess.py │ ├── rotation.py │ ├── tools.py │ └── visualise.py │ ├── split.py │ ├── test_generated_data.py │ └── uavhuman_pose_dataset.py ├── feeders ├── __init__.py ├── feeder.py └── tools.py ├── graph ├── __init__.py ├── kinetics.py ├── ntu_rgb_d.py ├── tools.py └── uav.py ├── main.py ├── model ├── Temporal_shift │ ├── cuda │ │ ├── __init__.py │ │ ├── setup.py │ │ ├── shift.py │ │ ├── shift_cuda.cpp │ │ └── shift_cuda_kernel.cu │ ├── demo.py │ ├── readme.txt │ └── run.sh ├── __init__.py ├── aagcn.py ├── agcn.py ├── graph ├── model ├── module_cau.py ├── module_ta.py ├── shift_gcn.py └── tegcn.py ├── runs └── runs_here ├── scripts ├── EVAL_V1.sh ├── EVAL_V2.sh ├── TRAIN_V1.sh └── TRAIN_V2.sh └── work_dir └── work_dir_here /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/README.md -------------------------------------------------------------------------------- /config/uav-cross-subjectv1/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/config/uav-cross-subjectv1/test.yaml -------------------------------------------------------------------------------- /config/uav-cross-subjectv1/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/config/uav-cross-subjectv1/train.yaml -------------------------------------------------------------------------------- /config/uav-cross-subjectv1/train_shiftgcn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/config/uav-cross-subjectv1/train_shiftgcn.yaml -------------------------------------------------------------------------------- /config/uav-cross-subjectv2/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/config/uav-cross-subjectv2/test.yaml -------------------------------------------------------------------------------- /config/uav-cross-subjectv2/train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/config/uav-cross-subjectv2/train.yaml -------------------------------------------------------------------------------- /config/uav-cross-subjectv2/train_shiftgcn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/config/uav-cross-subjectv2/train_shiftgcn.yaml -------------------------------------------------------------------------------- /data/data_here: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_gen/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_gen/gen_bone_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/gen_bone_data.py -------------------------------------------------------------------------------- /data_gen/gen_motion_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/gen_motion_data.py -------------------------------------------------------------------------------- /data_gen/kinetics_gendata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/kinetics_gendata.py -------------------------------------------------------------------------------- /data_gen/merge_joint_bone_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/merge_joint_bone_data.py -------------------------------------------------------------------------------- /data_gen/ntu_gendata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/ntu_gendata.py -------------------------------------------------------------------------------- /data_gen/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/preprocess.py -------------------------------------------------------------------------------- /data_gen/rotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/rotation.py -------------------------------------------------------------------------------- /data_gen/uavhumanposetools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/uavhumanposetools/README.md -------------------------------------------------------------------------------- /data_gen/uavhumanposetools/generate_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/uavhumanposetools/generate_data.py -------------------------------------------------------------------------------- /data_gen/uavhumanposetools/pose_data_tools/graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/uavhumanposetools/pose_data_tools/graph.py -------------------------------------------------------------------------------- /data_gen/uavhumanposetools/pose_data_tools/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/uavhumanposetools/pose_data_tools/preprocess.py -------------------------------------------------------------------------------- /data_gen/uavhumanposetools/pose_data_tools/rotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/uavhumanposetools/pose_data_tools/rotation.py -------------------------------------------------------------------------------- /data_gen/uavhumanposetools/pose_data_tools/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/uavhumanposetools/pose_data_tools/tools.py -------------------------------------------------------------------------------- /data_gen/uavhumanposetools/pose_data_tools/visualise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/uavhumanposetools/pose_data_tools/visualise.py -------------------------------------------------------------------------------- /data_gen/uavhumanposetools/split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/uavhumanposetools/split.py -------------------------------------------------------------------------------- /data_gen/uavhumanposetools/test_generated_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/uavhumanposetools/test_generated_data.py -------------------------------------------------------------------------------- /data_gen/uavhumanposetools/uavhuman_pose_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/data_gen/uavhumanposetools/uavhuman_pose_dataset.py -------------------------------------------------------------------------------- /feeders/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/feeders/__init__.py -------------------------------------------------------------------------------- /feeders/feeder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/feeders/feeder.py -------------------------------------------------------------------------------- /feeders/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/feeders/tools.py -------------------------------------------------------------------------------- /graph/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/graph/__init__.py -------------------------------------------------------------------------------- /graph/kinetics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/graph/kinetics.py -------------------------------------------------------------------------------- /graph/ntu_rgb_d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/graph/ntu_rgb_d.py -------------------------------------------------------------------------------- /graph/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/graph/tools.py -------------------------------------------------------------------------------- /graph/uav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/graph/uav.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/main.py -------------------------------------------------------------------------------- /model/Temporal_shift/cuda/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/Temporal_shift/cuda/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/Temporal_shift/cuda/setup.py -------------------------------------------------------------------------------- /model/Temporal_shift/cuda/shift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/Temporal_shift/cuda/shift.py -------------------------------------------------------------------------------- /model/Temporal_shift/cuda/shift_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/Temporal_shift/cuda/shift_cuda.cpp -------------------------------------------------------------------------------- /model/Temporal_shift/cuda/shift_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/Temporal_shift/cuda/shift_cuda_kernel.cu -------------------------------------------------------------------------------- /model/Temporal_shift/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/Temporal_shift/demo.py -------------------------------------------------------------------------------- /model/Temporal_shift/readme.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/Temporal_shift/readme.txt -------------------------------------------------------------------------------- /model/Temporal_shift/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/Temporal_shift/run.sh -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/aagcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/aagcn.py -------------------------------------------------------------------------------- /model/agcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/agcn.py -------------------------------------------------------------------------------- /model/graph: -------------------------------------------------------------------------------- 1 | ../graph -------------------------------------------------------------------------------- /model/model: -------------------------------------------------------------------------------- 1 | ../model -------------------------------------------------------------------------------- /model/module_cau.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/module_cau.py -------------------------------------------------------------------------------- /model/module_ta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/module_ta.py -------------------------------------------------------------------------------- /model/shift_gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/shift_gcn.py -------------------------------------------------------------------------------- /model/tegcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/model/tegcn.py -------------------------------------------------------------------------------- /runs/runs_here: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/EVAL_V1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/scripts/EVAL_V1.sh -------------------------------------------------------------------------------- /scripts/EVAL_V2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/scripts/EVAL_V2.sh -------------------------------------------------------------------------------- /scripts/TRAIN_V1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/scripts/TRAIN_V1.sh -------------------------------------------------------------------------------- /scripts/TRAIN_V2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xieyulai/TE-GCN/HEAD/scripts/TRAIN_V2.sh -------------------------------------------------------------------------------- /work_dir/work_dir_here: -------------------------------------------------------------------------------- 1 | --------------------------------------------------------------------------------