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

8.04k stars 1.98k forks source link

Can I finetune Alphapose on custom keypoint dataset? #1175

Open RohaanA opened 1 year ago

RohaanA commented 1 year 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 1 year 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.