fix crop

tensorpack/dataflow/imgaug/crop.py

@@ -4,6 +4,7 @@
 # Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
 from .base import ImageAugmentor
+import cv2
 
 __all__ = ['RandomCrop', 'CenterCrop', 'Resize']
 
@@ -42,11 +43,11 @@ class Resize(ImageAugmentor):
     def __init__(self, shape):
         """
         Args:
-            shape: (w, h)
+            shape: (h, w)
         """
         self._init(locals())
 
     def _augment(self, img):
         img.arr = cv2.resize(
-            img.arr, self.shape,
+            img.arr, self.shape[::-1],
             interpolation=cv2.INTER_CUBIC)

tensorpack/models/image_sample.py

@@ -46,6 +46,7 @@ def ImageSample(template, mapping):
     mapping: bxh2xw2x2  (y, x) real-value coordinates
     Return: bxh2xw2xc
     """
+    assert template.get_shape().ndims == 4 and mapping.get_shape().ndims == 4
     mapping = tf.maximum(mapping, 0.0)
     tf.check_numerics(mapping, "mapping")
     lcoor = tf.cast(mapping, tf.int32)  # floor