Skip to content

Swadesh13/STGCN-Tf2

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

STGCN-Tf2

Spatio-Temporal Graph Convolutional Neural Network

This repository contains code updated for Tensorflow Version 2 (TF2 + Keras) from the original repository. The Keras component is very limited. For TF2 only, check the branch named 'tf2-only'.

Other than code conversion from TFv1 to TFv2, certain additional changes have been made in the data loading mechanism, inference shapes not matching and a model update (Ko -= 2*(Kt - 1)). This code runs slower than the original version.

For my project, the most updated code is present in another repository that has additional changes and also, a number of added models to compare various combinations of Spatial-GCN and Temporal-GCN layers.

About

TF2 updated version of the original code.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages