Skip to content
Snippets Groups Projects
Commit c68890db authored by Jiangmiao Pang's avatar Jiangmiao Pang Committed by Kai Chen
Browse files

Fix IoU sampling (#1784)

parent b7894cbd
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py
+ 2
− 0
View file @ c68890db
...@@ -98,6 +98,8 @@ class IoUBalancedNegSampler(RandomSampler): ...@@ -98,6 +98,8 @@ class IoUBalancedNegSampler(RandomSampler):
floor_set = set() floor_set = set()
iou_sampling_set = set( iou_sampling_set = set(
np.where(max_overlaps > self.floor_thr)[0]) np.where(max_overlaps > self.floor_thr)[0])
# for sampling interval calculation
self.floor_thr == 0
floor_neg_inds = list(floor_set & neg_set) floor_neg_inds = list(floor_set & neg_set)
iou_sampling_neg_inds = list(iou_sampling_set & neg_set) iou_sampling_neg_inds = list(iou_sampling_set & neg_set)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment