How can I change the number of keypoints about the dataset?

[YHDang]

Hello, thanks for your excellent research! I tried to reproduce the YOLO-Pose, but met a problem. I want to train the model on the CrowdPose dataset. When I generate the label files with the yolo format (train.txt, val.txt, and test.txt), I found that the number of keypoints must keep 17 (that is the same as the COCO). If the number of keypoints is changed, *.cache files will error. Could you give me some suggestions that how and where I can change the number of keypoints in the source code?

[akashAD98]

@YHDang do you know how to generate keypoints for custom data?

[il-nietos]

@YHDang were you able to run the model with a different number of keypoints?