Skip to content

Spatial Temporal Graph Convolutional Networks for Skeleton-Based Action Recognition

Notifications You must be signed in to change notification settings

yongqyu/st-gcn-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Human Action Recognition

This code is implemented based on Spatial Temporal Graph Convolutional Networks for Skeleton-Based Action Recognition.
And this code only supports the Florence 3D action dataset.
Please download from here at ./dataset/

Environments

  • python 3.6
  • pytorch 1.0

Run

python main.py

File Details

File Name Description
preprocess.py Preprocess the Florence 3D action dataset. Each frame is unified into 32 frames, separated by train / valid / test and dropped into a file.
config.py Hyperparameter and data path setting
main.py Execution File. Loads data and models, and performs training and testing.
model.py Defines the General Graph Convolutional Network (GGCN) class. Construct an adjacency matrix with three consecutive graphs, and call several layers.
layer.py Defines the graph convolution, standard convolution and classifier layer.
metric.py Defines the accuracy function

About

Spatial Temporal Graph Convolutional Networks for Skeleton-Based Action Recognition

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages