| from owlv2_helper_functions import get_iou, boxes_filter | |
| boxes = [ | |
| (128.56, 4.57, 732.52, 476.05), | |
| (569.65, 185.71, 740.31, 244.76), | |
| (569.65, 185.71, 740.31, 244.76), | |
| (569.65, 185.71, 740.31, 244.76), | |
| (101.99, 99.00, 720.12, 88.63), | |
| ] | |
| scores = [1.0, 0.99, 0.89, 1.0, 0.99] | |
| instances = ['cat', 'dog', 'dog', 'tiger', 'cat'] | |
| pred_bboxes, pred_scores, instances = boxes_filter(boxes, scores, instances) | |
| print(pred_bboxes) | |
| print(pred_scores) | |
| print(instances) |