fix augmentor bug

tensorpack/dataflow/imgaug/base.py

@@ -77,6 +77,9 @@ class AugmentorList(ImageAugmentor):
         raise RuntimeError("Cannot simply get parameters of a AugmentorList!")
 
     def _augment_return_params(self, img):
+        assert img.ndim in [2, 3], img.ndim
+        img = img.astype('float32')
+
         prms = []
         for a in self.augs:
             img, prm = a._augment_return_params(img)

tensorpack/dataflow/imgaug/deform.py

@@ -88,7 +88,6 @@ class GaussianDeform(ImageAugmentor):
 
     def _augment(self, img, v):
         grid = self.grid + np.dot(self.gws, v)
-        print(grid)
         return np_sample(img, grid)
 
     def _fprop_coord(self, coord, param):

tensorpack/dataflow/imgaug/noname.py

@@ -54,7 +54,7 @@ class MapImage(ImageAugmentor):
         self.func = func
 
     def _augment(self, img, _):
-        img = self.func(img)
+        return self.func(img)
 
 
 class Resize(ImageAugmentor):
@@ -66,6 +66,6 @@ class Resize(ImageAugmentor):
         self._init(locals())
 
     def _augment(self, img, _):
-        img.arr = cv2.resize(
-            img.arr, self.shape[::-1],
+        return cv2.resize(
+            img, self.shape[::-1],
             interpolation=cv2.INTER_CUBIC)

tensorpack/dataflow/imgaug/paste.py

@@ -67,7 +67,8 @@ class CenterPaste(ImageAugmentor):
         w0 = int((self.background_shape[1] - img_shape[1]) * 0.5)
         background[h0:h0+img_shape[0], w0:w0+img_shape[1]] = img
         img = background
-        if img.coords:
-            raise NotImplementedError()
+        return img
 
+    def _fprop_coord(self, coord, param):
+        raise NotImplementedError()