1.5 KiB

Raw Permalink Blame History

Just Dance - A Simple Implementation

^{Try it on OpenXLab}

This project presents a dance scoring system based on RTMPose. Users can compare the similarity between two dancers in different videos: one referred to as the "teacher video" and the other as the "student video."

Here are examples of the output dance comparison:

Usage

Jupyter Notebook

We provide a Jupyter Notebook just_dance_demo.ipynb that contains the complete process of dance comparison. It includes steps such as video FPS adjustment, pose estimation, snippet alignment, scoring, and the generation of the merged video.

CLI tool

Users can simply run the following command to generate the comparison video:

python process_video.py ${TEACHER_VIDEO} ${STUDENT_VIDEO}

Gradio

Users can also utilize Gradio to build an application using this system. We provide the script app.py. This application supports webcam input in addition to existing videos. To build this application, please follow these two steps:

Install Gradio
```
pip install gradio
```
Run the script app.py
```
python app.py
```

1.5 KiB Raw Permalink Blame History