From 7636de2df1c56a8b7b5306cf5ead05cf4d0b2721 Mon Sep 17 00:00:00 2001
From: Dahun Kim <mcahny@github.com>
Date: Tue, 31 Aug 2021 20:48:51 +0900
Subject: [PATCH] minor fix

---
 configs/oln_box/oln_box.py               | 2 +-
 mmdet/models/dense_heads/oln_rpn_head.py | 3 ---
 2 files changed, 1 insertion(+), 4 deletions(-)

diff --git a/configs/oln_box/oln_box.py b/configs/oln_box/oln_box.py
index a6f4cc6..1ba5c41 100644
--- a/configs/oln_box/oln_box.py
+++ b/configs/oln_box/oln_box.py
@@ -129,7 +129,7 @@ model = dict(
             nms_pre=2000,
             nms_post=2000,
             max_num=2000,
-            nms_thr=0.7,
+            nms_thr=0.0,  # No nms
             min_bbox_size=0),
         rcnn=dict(
             score_thr=0.0,
diff --git a/mmdet/models/dense_heads/oln_rpn_head.py b/mmdet/models/dense_heads/oln_rpn_head.py
index 4137101..f65f0b9 100644
--- a/mmdet/models/dense_heads/oln_rpn_head.py
+++ b/mmdet/models/dense_heads/oln_rpn_head.py
@@ -691,9 +691,6 @@ class OlnRPNHead(RPNHead):
 
         # No NMS:
         dets = torch.cat([proposals, scores.unsqueeze(1)], 1)
-        # else:
-        #     dets, keep = batched_nms(proposals, scores, ids, nms_cfg)
-        #     return dets[:cfg.nms_post]
         
         return dets