README.md

Track multiple classes

An example of how to use Norfair to track objects from multiple classes using both keypoints and bounding boxes. This example is based on OpenPose and YOLOv5.

Instructions

1. Build and run the Docker container with ./run_gpu.sh.

2. Copy a video to the src folder.

3. Within the container, run with the default parameters:

   python demo.py <video>.mp4

For additional settings, you may display the instructions using python demo.py --help.

Alternatively, the example can be executed entirely within keypoints_bounding_boxes_demo.ipynb.

Explanation

This example aims at showing 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.

keypoints_bounding_boxes.mp4