christhornton555 / machine_vision

0 stars 0 forks source link

Add Skeleton Tracking #2

Closed christhornton555 closed 1 month ago

christhornton555 commented 1 month ago

Add tracking points to a detected person's body if they fill a decent amount of the frame, and join them into a skeleton

christhornton555 commented 1 month ago

Tracking's now working, and I've applied the standard OpenPose colours to the bones too. Mostly. I guess the yolov8n-pose.pt model doesn't have a neck keypoint like OpenPose, so I might try fiddling that before I close this issue

christhornton555 commented 1 month ago

Done a quick & dirty neck approximation. It could maybe do with a little proportional pulling towards centre-mass, but it'll do for now. However, the keypoints are just using their bone colours, and should really pull their brighter colours. Also, all these repetitive bits could be done more neatly. Get them working first, then have a think about that.

christhornton555 commented 1 month ago

Done some refactoring to put more static stuff in the config file, and added the correct keypoint colours. This issue is done.