RizwanMunawar / yolov7-pose-estimation

YOLOv7 Pose estimation using OpenCV, PyTorch
GNU General Public License v3.0
335 stars 77 forks source link

Multi class detection with variable number of keypoints (nkpt) #21

Open adrirg9412 opened 1 year ago

adrirg9412 commented 1 year ago

Hi,

I am researching how to custom train yolov7-pose on my own dataset with 5 different classes but one of them is person class which have 17 keypoints and in the other ones we want only one keypoint or none keypoint,

is it possible? How have I to modify .cfg file to do so?

Thank you.

crazyai-kby commented 1 year ago

Hello @adrirg9412 Did you get it working now?