DuyNguDao / ST-GCN-Pytorch

8 stars 2 forks source link

HUMAN ACTION RECOGNITION WITH ST-GCN

Introduction

This is project train human action recognition depend on skeleton with ST-GCN. Skeleton extracted with model YOLOv7-Pose

Dev

Member:
- Dao Duy Ngu
- Le Van Thien

Usage

Install package

git clone https://github.com/DuyNguDao/ST-GCN-Pytorch.git
cd ST-GCN-Pytorch
conda create --name human_action python=3.8
pip install -r requirements.txt

Dataset

This is dataset include with 7 class: sitting, sit down, standing, walking, stand up, lying down, fall down

video

video

pose

pose using YOLOv7-Pose

pose yolov3

pose using YOLOv3 + Alphapose

pose yolov7

Training

python train.py

Test

python test.py

Acknowledgements