Merge pull request #104 from mapbox/assert-resolution

Assert image resolution has to be divisible by 32 for resnet
mapbox · Aug 10, 2018 · 495edfb · 495edfb
2 parents d33ae08 + d65190b
commit 495edfb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/robosat/unet.py b/robosat/unet.py
@@ -116,6 +116,8 @@ def forward(self, x):
         Returns:
           The networks output tensor.
         """
+        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)