Skip to content

Commit a64c6a5

Browse files
fix lint
1 parent e9e350e commit a64c6a5

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

mmdet3d/models/task_modules/samplers/iou_neg_piecewise_sampler.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
# Copyright (c) OpenMMLab. All rights reserved.
22
import math
3+
34
import torch
45

56
from mmdet3d.registry import TASK_UTILS
@@ -77,9 +78,9 @@ def _sample_neg(self, assign_result, num_expected, **kwargs):
7778
# the same number in the current piece.
7879
piece_expected_num = min(
7980
num_expected,
80-
math.ceil(
81-
num_expected *
82-
self.neg_piece_fractions[piece_inds]) + extend_num)
81+
math.ceil(num_expected *
82+
self.neg_piece_fractions[piece_inds]) +
83+
extend_num)
8384
min_iou_thr = self.neg_iou_thr[piece_inds + 1]
8485
max_iou_thr = self.neg_iou_thr[piece_inds]
8586
piece_neg_inds = torch.nonzero(

0 commit comments

Comments
 (0)