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

update floor_thr in iou sampling (#1800)

parent d6a592d7
No related branches found
No related tags found
No related merge requests found
......@@ -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)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment