Can I finetune Alphapose on custom keypoint dataset?

MVIG-SJTU / AlphaPose

Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System

http://mvig.org/research/alphapose.html

Other

7.83k stars 1.96k forks source link

Can I finetune Alphapose on custom keypoint dataset? #1175

Open RohaanA opened 10 months ago

RohaanA commented 10 months ago

Hi,

I am looking to finetune alphapose on my own dataset but it has custom keypoints (Human joints + Tennis Racket keypoints)

Can I do this in Alphapose?

HaoyiZhu commented 9 months ago

Hi, you just need to write your own dataset (follow our COCO dataset for example), and modify the related parts in config such as dataset name and keypoint number, etc.