diff --git a/mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py b/mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py index 5e3ffc3e914f09b82b301eb019425f798e72158b..d9239e0708de207254c832dd50850aca03f103f0 100644 --- 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)