File size: 487 Bytes
742a3d1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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) |