Assert image resolution has to be divisible by 32 for resnet in fpn

mapbox · Oct 15, 2018 · 8876768 · 8876768
1 parent 7d8bed5
commit 8876768
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/robosat/fpn.py b/robosat/fpn.py
@@ -46,6 +46,9 @@ def __init__(self, num_filters=256, pretrained=True):
     def forward(self, x):
         # Bottom-up pathway, from ResNet
 
+        size = x.size()
+        assert size[-1] % 32 == 0 and size[-2] % 32 == 0, "image resolution has to be divisible by 32 for resnet"
+
         enc0 = self.resnet.conv1(x)
         enc0 = self.resnet.bn1(enc0)
         enc0 = self.resnet.relu(enc0)