Non_max_suppression and the argument of classes

[ai_curious]

Lots of prior discussion of this topic. Here is one…

W3: Question with non max suppression and classes

More than you ever wanted to know about YOLO and NMS. NMS prunes duplicates Non-max suppression is used in YOLO to suppress predictions likely to be of the same object. It ignores the class predictions and uses a Jaccard Similarity Coefficient, aka IOU, and uses only the two predicted center locations and shapes transformed into bounding boxes. If the IOU between two predicted bounding boxes is sufficiently high then they are treated as if they are same object and only one of the two, the on…

Think crisply about what the set of boxes output by the YOLO CNN represents, what non-max suppression accomplishes without considering classes, and what the inclusion of classes would change. What is the benefit of including class at that step? What is the cost? Let us know?