diff --git a/mmdet/models/roi_extractors/single_level.py b/mmdet/models/roi_extractors/single_level.py index 6a1c1e7292f3d71642c9e93b7194c1378f398ce2..8731126440367178265209c3e1d3fca473556c02 100644 --- a/mmdet/models/roi_extractors/single_level.py +++ b/mmdet/models/roi_extractors/single_level.py @@ -54,10 +54,10 @@ class SingleRoIExtractor(nn.Module): def map_roi_levels(self, rois, num_levels): """Map rois to corresponding feature levels by scales. - - scale < finest_scale: level 0 - - finest_scale <= scale < finest_scale * 2: level 1 - - finest_scale * 2 <= scale < finest_scale * 4: level 2 - - scale >= finest_scale * 4: level 3 + - scale < finest_scale * 2: level 0 + - finest_scale * 2 <= scale < finest_scale * 4: level 1 + - finest_scale * 4 <= scale < finest_scale * 8: level 2 + - scale >= finest_scale * 8: level 3 Args: rois (Tensor): Input RoIs, shape (k, 5).