Pytorch implementation of skeleton transformer module
Project description
Skeleton Transformer Module in Pytorch
Pytorch 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.
pip install skeletorch
How to use
Module
skeletorch.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))
input_kpts_num: Number of joints in original keypoints (mentioned as 'N' in the paper)
output_dim: Dimentions of output (mentioned as 'M' in the paper)
Input:
x: 2-dimentional tensor of shape (timesteps, input_kpts_num*kpts_dim)
Example
import torch
from skeletorch import SkeletonTransformer
# parameters
timesteps = 20
kpts_dim = 3
input_kpts_num = 17
output_dim = 10
# input (the size is (20, 51) in this example)
x = torch.Tensor(torch.randn(timesteps, kpts_dim*input_kpts_num))
# make layer
layer = SkeletonTransformer(timesteps, kpts_dim, input_kpts_num, output_dim)
layer(x)
Of course you can use this module in your Pytorch network.
Keras implementation
We also have Keras 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.
Source Distribution
Built Distribution
Hashes for skeletorch-0.1.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d24c9c1cd03ad25f3f512d1cbc57ffde14e71d3efe8b63d3b7dbeb21f2a22679 |
|
MD5 | e5e2c44e7a7d90b98ae89bf32dcce84a |
|
BLAKE2b-256 | 2c5fcfe8e7b2c996c75e34db475fc45e0160ebdec6b410a104b2fdff3001ee62 |