RizwanMunawar / yolov7-pose-estimation

YOLOv7 Pose estimation using OpenCV, PyTorch
GNU General Public License v3.0
339 stars 78 forks source link

Keypoints extraction #19

Open Avishaek opened 1 year ago

Avishaek commented 1 year ago

How do i extract keypoints file . I intend to extract the keypoints for each ID and use these keypoints for action recognition using a classifier

RizwanMunawar commented 1 year ago

@Avishaek! you can store key points Information in some text files, and then use them for further tasks. For more info, you can look at the code line mentioned below.

https://github.com/RizwanMunawar/yolov7-pose-estimation/blob/60e65b2ec57e54c05f63c3917efe48db06ef0c71/pose-estimate.py#L96

Avishaek commented 1 year ago

Will it give unique ID for each skeleton over the complete video

RizwanMunawar commented 1 year ago

Will it give a unique ID for each skeleton over the complete video

In most cases, yes

adisomani2003 commented 1 year ago

@Avishaek were you able to do this, and store the keypoints frame by frame?