Skip to content

Commit af866c4

Browse files
gkioxarifacebook-github-bot
authored andcommitted
fix to pool with scale 1
Summary: Small fix to pool at scale 1 in ROIPooler Reviewed By: ppwwyyxx Differential Revision: D22809193 fbshipit-source-id: 2512b47354a85bca6590496173bd6706e274e3cc
1 parent 4c762c8 commit af866c4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

detectron2/modeling/poolers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ def __init__(
171171
assert (
172172
len(scales) == self.max_level - self.min_level + 1
173173
), "[ROIPooler] Sizes of input featuremaps do not form a pyramid!"
174-
assert 0 < self.min_level and self.min_level <= self.max_level
174+
assert 0 <= self.min_level and self.min_level <= self.max_level
175175
self.canonical_level = canonical_level
176176
assert canonical_box_size > 0
177177
self.canonical_box_size = canonical_box_size

0 commit comments

Comments
 (0)