-
Notifications
You must be signed in to change notification settings - Fork 63
Open
Labels
Description
In our detection_metrics module, compute_iou_matrix currently uses a Python double for loop to fill a matrix of IoU values, which is slow for large datasets.
# current (slow)
for i, pb in enumerate(pred_boxes):
for j, gb in enumerate(gt_boxes):
iou_matrix[i, j] = compute_iou(pb, gb)Deliverables: rewrite said function using fully vectorized NumPy broadcasting to compute all pairwise IoUs at once and make sure that the new implementation still yields the exact same results
Reactions are currently unavailable