[GRASS-SVN] r74151 - grass-addons/grass7/imagery/i.ann.maskrcnn/maskrcnnlib

[svn_grass at osgeo.org]

Author: pesekon2
Date: 2019-03-04 04:26:15 -0800 (Mon, 04 Mar 2019)
New Revision: 74151

Modified:
   grass-addons/grass7/imagery/i.ann.maskrcnn/maskrcnnlib/model.py
Log:
i.ann.maskrcnn: fix multiple anchors with same IoU

Modified: grass-addons/grass7/imagery/i.ann.maskrcnn/maskrcnnlib/model.py
===================================================================
--- grass-addons/grass7/imagery/i.ann.maskrcnn/maskrcnnlib/model.py	2019-03-03 23:29:56 UTC (rev 74150)
+++ grass-addons/grass7/imagery/i.ann.maskrcnn/maskrcnnlib/model.py	2019-03-04 12:26:15 UTC (rev 74151)
@@ -1502,8 +1502,8 @@
     anchor_iou_max = overlaps[np.arange(overlaps.shape[0]), anchor_iou_argmax]
     rpn_match[(anchor_iou_max < 0.3) & (no_crowd_bool)] = -1
     # 2. Set an anchor for each GT box (regardless of IoU value).
-    # TODO: If multiple anchors have the same IoU match all of them
-    gt_iou_argmax = np.argmax(overlaps, axis=0)
+    # If multiple anchors have the same IoU match all of them
+    gt_iou_argmax = np.argwhere(overlaps == np.max(overlaps, axis=0))[:,0]
     rpn_match[gt_iou_argmax] = 1
     # 3. Set anchors with high overlap as positive.
     rpn_match[anchor_iou_max >= 0.7] = 1