Add demo tracking keypoints and bounding boxes using OpenPose and YOLOv5 #99
Conversation
|
|
||
| ## Explanation | ||
|
|
||
| This example aims to show the possibilities that Norfair offers to track objects from multiple classes using a single `Tracker` instance. Also the example shows Norfair's ability to draw keypoints and bounding boxes for different types of objects. |
There was a problem hiding this comment.
This example aims to show the possibilities -> This example aims at showing the possibilities
| distances = np.linalg.norm(detected_pose.points - tracked_pose.estimate, axis=1) | ||
| match_num = np.count_nonzero( | ||
| (distances < KEYPOINT_DIST_THRESHOLD) | ||
| * (detected_pose.scores > DETECTION_THRESHOLD) | ||
| * (tracked_pose.last_detection.scores > DETECTION_THRESHOLD) | ||
| ) | ||
| return 1 / (1 + match_num) |
There was a problem hiding this comment.
Would be slightly more readable under an else statement, as we are choosing between 2 distinct options.
| ) | ||
| KEYPOINT_DIST_THRESHOLD = video.input_height / 40 | ||
|
|
||
| for i, frame in enumerate(video): |
There was a problem hiding this comment.
I'd personally not support skipping frames in this demo. There are a lot of things going on, and we want to make each demo show off a small, easy to grasp, subset of Norfair's features. So anything that makes the demo shorter and simpler is better (except not tracking 2 classes 😂).
|
Just saw that that video is the input video for the demo. We should probably not version it, and store it somewhere so that users can download it manually. Maybe add a wget command to the demo or something like that. Storing it in github would probably be ideal. Same goes for the gif. We can chat about this on slack later. |
facundo-lezama
force-pushed
the
demo-keypoints-bboxes
branch
from
e821007
to
b2a33f9
Compare
|
Made the requested changes. Regarding the input video I uploaded it by editing the README file from the web UI. |
facundo-lezama
force-pushed
the
demo-keypoints-bboxes
branch
from
b2a33f9
to
d14725b
Compare
facundo-lezama
force-pushed
the
demo-keypoints-bboxes
branch
from
d14725b
to
af44da4
Compare
Add demo tracking keypoints and bounding boxes using OpenPose and YOLOv5
Add demo tracking keypoints and bounding boxes using OpenPose and YOLOv5
This PR builds a demo to show the possibilities that Norfair offers to track objects from multiple classes using a single
Trackerinstance. Also the example shows Norfair's ability to draw keypoints and bounding boxes for different types of objects.