sort
sort copied to clipboard
Added sort_inter_class.py
When you have lots of objects belonging to a different class, this will only run sort on objects of the same class. This can be combined with YOLO to track several different objects. What it essentially does is, in the IOU matrix, it assigns a value to 0 to any matched bounding boxes if the class of the two bounding boxes is different.
So, for example, if the bounding box of the Person class matches that of the Bicycle class, the final IoU score will still be made 0 because both are two different classes. While, if the two bounding boxes are of the same class, this will not be done and the original IoU will be used as-is.
hey man, great stuff. iou batch i think should be castto float32 to avoid the range issue, no?
@haviduck I didn't encounter any range issues, can you do a test run and check if there is a problem?
been so long and i dont have a vanilla setup to test. my current code throws index errors, but i csn look into it.
**Update Yeah, it works fine. empty detections threw me for a loop. good job :)
@abewley @sodabeta7 if there are no issues with this commit, can you please add it to the repo? Thanks