Skip to content

KPeng9510/Trans4SOAR

Repository files navigation

Trans4SOAR: Delving Deep into One-Shot Skeleton-based Action Recognition with Diverse Occlusions

Introduction of our work

In this work, we focus on skeleton-based one-shot action recognition with diverse occlusion scenarios. In order to get the realistic synthesized occluded dataset for NTU-120, NTU-60 and Toyota Smart Home, please send the license agreement of the original datasets to pengkunyu1013@gmail.com. After checking the datasets will be sent via email. Thank you for your interests!

plot

Environment

The required packages are listed in the environment.yml, please create conda env accordingly.

Dataloader

The dataloaders can be found in the dataloader folder, before the running of the code please change the corresponding path accordingly in the dataloader.

How to run the code

First, execute, export DATASET_FOLDER="$(pwd)/data" Then, python train_original.py If you are using Trans4SOAR model please make sure you use one gpu for training.

Dataset download link (RE Occluded NTU120)

https://drive.google.com/drive/folders/1E6O9S5Z-ZWTg6iHqZnTM-o-oX9Ica9Bk?usp=share_link

Dataset download link (Occluded NTU60)

In uploading

Dataset download link (Occluded ToyotaSmartHome)

In uploading

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •  

Languages