Keras implementation of skeleton transformer module
Project description
Skeleton Transformer in Keras
Keras implementation of "skeleton transformer module", which is mentioned in Skeleton-based Action Recognition with Convolutional Neural Networks.
Install
You can install this module from PyPI as skeraton
.
pip install skeraton
How to use
Module
skeraton.SkeletonTransformer
Parameters:
All parameters are required.
timesteps: Timesteps of input time-series data (equal to number of frames, mentioned as 'T' in the paper)
kpts_dim: Dimentions of keypoints (usually 2 (x, y) or 3 (x, y, z))
output_dim: Dimentions of output (mentioned as 'M' in the paper)
Input:
x: 3-dimentional tensor of shape (batchsize, timesteps, kpts_dim*N) where N is number of joints
Pytorch implementation
We also have Pytorch implementation of this module:
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.