Update bbox_transform.py

winday00 · Jul 18, 2019 · 2eceb34 · 2eceb34
1 parent 358ceca
commit 2eceb34
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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.