fix to pool with scale 1
Summary: Small fix to pool at scale 1 in ROIPooler
@@ -171,7 +171,7 @@ class ROIPooler(nn.Module):
assert (
len(scales) == self.max_level - self.min_level + 1
), "[ROIPooler] Sizes of input featuremaps do not form a pyramid!"
- assert 0 < self.min_level and self.min_level <= self.max_level
+ assert 0 <= self.min_level and self.min_level <= self.max_level
self.canonical_level = canonical_level
assert canonical_box_size > 0
self.canonical_box_size = canonical_box_size
