movenet.pytorch icon indicating copy to clipboard operation
movenet.pytorch copied to clipboard

Train with more than 17 keypoints

Open Darkaido opened this issue 2 years ago • 1 comments

Hii, i want to train movenet with more than 17 keypoints. so what changes i have to do for training.

Darkaido avatar Jan 05 '23 09:01 Darkaido

Hi, maybe you should do this:

  1. change data generate script: scripts/make_coco_data_17keypooints.py to your number
  2. change model head structure, such as output channel number, etc.
  3. change the way how loss computed
  4. also be careful to some data aware augmentation, like issue36

fire717 avatar Jan 06 '23 01:01 fire717