sort icon indicating copy to clipboard operation
sort copied to clipboard

Added sort_inter_class.py

Open sarimmehdi opened this issue 3 years ago • 4 comments

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.

sarimmehdi avatar Aug 17 '21 13:08 sarimmehdi

hey man, great stuff. iou batch i think should be castto float32 to avoid the range issue, no?

haviduck avatar Aug 29 '21 23:08 haviduck

@haviduck I didn't encounter any range issues, can you do a test run and check if there is a problem?

sarimmehdi avatar Nov 11 '21 19:11 sarimmehdi

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 :)

haviduck avatar Nov 11 '21 22:11 haviduck

@abewley @sodabeta7 if there are no issues with this commit, can you please add it to the repo? Thanks

sarimmehdi avatar Jan 29 '22 22:01 sarimmehdi