search

TexasInstruments / edgeai-yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite. Forked from https://ultralytics.com/yolov5
https://github.com/TexasInstruments/edgeai
GNU General Public License v3.0
654 stars 120 forks source link

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

Open YHDang opened 2 years ago

YHDang commented 2 years ago

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 commented 2 years ago

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

il-nietos commented 1 year ago

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