diff --git a/lib/model/rpn/bbox_transform.py b/lib/model/rpn/bbox_transform.py index 450de0f56..79ded21c0 100644 --- a/lib/model/rpn/bbox_transform.py +++ b/lib/model/rpn/bbox_transform.py @@ -245,7 +245,8 @@ def bbox_overlaps_batch(anchors, gt_boxes): torch.max(boxes[:,:,:,1], query_boxes[:,:,:,1]) + 1) ih[ih < 0] = 0 ua = anchors_area + gt_boxes_area - (iw * ih) - + + # Intersection (iw * ih) divided by Union (ua) overlaps = iw * ih / ua # mask the overlap here.