From c68890db5910eed4fc8ec2acf4cdf1426cb038e9 Mon Sep 17 00:00:00 2001
From: Jiangmiao Pang <pangjiangmiao@gmail.com>
Date: Tue, 10 Dec 2019 01:34:41 -0800
Subject: [PATCH] Fix IoU sampling (#1784)
---
mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py b/mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py
index 62431d6..5e3ffc3 100644
--- a/mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py
+++ b/mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py
@@ -98,6 +98,8 @@ class IoUBalancedNegSampler(RandomSampler):
floor_set = set()
iou_sampling_set = set(
np.where(max_overlaps > self.floor_thr)[0])
+ # for sampling interval calculation
+ self.floor_thr == 0
floor_neg_inds = list(floor_set & neg_set)
iou_sampling_neg_inds = list(iou_sampling_set & neg_set)
--
GitLab