update floor_thr in iou sampling (#1800)

277b5dca · Jiangmiao Pang · Kai Chen · d6a592d7 · 277b5dca
Commit 277b5dca authored Dec 13, 2019 by Jiangmiao Pang Committed by Kai Chen Dec 13, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py +1 -1

No files found.
--- a/mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py
+++ b/mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py
@@ -99,7 +99,7 @@ class IoUBalancedNegSampler(RandomSampler):
                iou_sampling_set = set(
                    np.where(max_overlaps > self.floor_thr)[0])
                # for sampling interval calculation
-                self.floor_thr == 0
+                self.floor_thr = 0

            floor_neg_inds = list(floor_set & neg_set)
            iou_sampling_neg_inds = list(iou_sampling_set & neg_set)