negarhdr / skeleton-based-action-recognition

This repository provides implementation of a baseline method and our proposed methods for efficient Skeleton-based Human Action Recognition.
Other
26 stars 6 forks source link

How can it be used for real time recognition? #2

Open lakpa-tamang9 opened 2 years ago

lakpa-tamang9 commented 2 years ago

I have some queries:

  1. How do you load your model for inference in real time?
  2. Can we use this for developing with my custom dataset?
negarhdr commented 2 years ago

Hi,

Thanks for your interest in my work. The code is integrated in OpenDR toolkit and a demo is available there: https://github.com/opendr-eu/opendr/blob/master/projects/perception/skeleton_based_action_recognition/demos/demo.py

In order to train the models on a custom-skeleton dataset, you need first to employ lightweight openpose method to extract body poses from the videos. We have a code for that in this directory: https://github.com/opendr-eu/opendr/blob/master/projects/perception/skeleton_based_action_recognition/demos/skeleton_extraction.py

lakpa-tamang9 commented 2 years ago

Thanks for your reply. I implemented the work on my custom dataset and ran real time recognition. However, the fps is very low at around 2 fps. In your demo i can see the fps is quite good around 15-18 fps. I have couple of questions. I hope you can clear my queries.

Thank you for your time. Cheers!!!