I want to use only 6 key points, not a 16 key point pre-trained model.

[NaJongHo]

I want to use only 6 key points, not a 16 key point pre-trained model.
So I ran the code with scratch.

but I keep getting errors like photos.

Any comments would be greatly appreciated.

[yuanyuanli85]

you need to make sure the training samples provide you the right shape.

[chojuahn]

It seems like the model is still expecting 16 key points. You should try searching hard-coded number '16' all over the code base and replace with 6