movenet.pytorch
movenet.pytorch copied to clipboard
Train with more than 17 keypoints
Hii, i want to train movenet with more than 17 keypoints. so what changes i have to do for training.
Hi, maybe you should do this:
- change data generate script:
scripts/make_coco_data_17keypooints.py
to your number - change model head structure, such as output channel number, etc.
- change the way how loss computed
- also be careful to some data aware augmentation, like issue36