Two-Stage Suppression for Multi-Class, Multi-Object Detection and Tracking Systems

A computer-implemented method for performing two-stage suppression of bounding boxes, the method comprising:

• receiving an image depicting a first object having a first class and a second object having a second class;
• generating preliminary bounding boxes for the first object and the second object;
• grouping the preliminary bounding boxes into a first group of preliminary bounding boxes having the first class and a second group of preliminary bounding boxes having the second class;
• selecting a first subset of bounding boxes by separately performing NMS on the first group of preliminary bounding boxes and the second group of preliminary bounding boxes to deduplicate bounding boxes from the first group and the second group;
• and
• selecting a second subset of bounding boxes by performing NMS on the first subset of bounding boxes, without regard to class.

A computer-implemented method for performing two-stage suppression of bounding boxes, the method comprising:

• receiving an image depicting multiple objects belonging to different classes;
• generating preliminary bounding boxes an image depicting multiple objects, wherein each of the preliminary bounding boxes comprises a size, location, class, and confidence score;
• performing a per-class NMS of the preliminary bounding boxes to select a subset of bounding boxes;
• performing a class-agnostic NMS of the subset of bounding boxes to select a set of filtered of bounding boxes;
• and
• based on the set of filtered bounding boxes, generating at least one of an enriched image or a video index.