Perform non maximum suppression for classes separately (request to extend code)

[schneider128k]

Right now the non maximum suppression is performed for all the bounding boxes for all the classes.

yolov3-tf2/yolov3_tf2/models.py

Line 175 in f38bb5a

def yolo_nms(outputs, anchors, masks, classes):

Would it be possible to add a flag that allows you to carry out the non maximum class separately for different classes?

I have a use case where a large object can sometimes have a smaller object attached to it. The problem is that the bounding box of the smaller object (if it is present) is always suppressed by the bounding object of the larger object.

I have found a hacky solution that works for me, but I think that it would be useful to a have a general solution.