fix imports

examples/ResNet/imagenet-resnet.py

@@ -16,7 +16,7 @@ from tensorpack.tfutils.symbolic_functions import *
 from tensorpack.tfutils.summary import *
 
 """
-Training code of ResNet on ImageNet. Work In Progress.
+Training code of Pre-Activation version of ResNet on ImageNet. Work In Progress.
 Top1 error is now about 0.5% higher than fb.resnet.torch.
 """
 
@@ -136,11 +136,9 @@ def get_data(train_or_test):
     image_std = np.array([0.229, 0.224, 0.225], dtype='float32')
 
     if isTrain:
-        class Resize(imgaug.ImageAugmentor):
-            def __init__(self):
-                self._init(locals())
-            def _augment(self, img, _):
-                # fbaug
+        def resize_func(img):
+            # crop 8%~100% of the original image
+            # See `Going Deeper with Convolutions` by Google.
             h, w = img.shape[:2]
             area = h * w
             for _ in range(10):
@@ -160,7 +158,7 @@ def get_data(train_or_test):
             return out
 
         augmentors = [
-            Resize(),
+            imgaug.MapImage(resize_func),
             imgaug.RandomOrderAug(
                 [imgaug.Brightness(30, clip=False),
                  imgaug.Contrast((0.8, 1.2), clip=False),

tensorpack/tfutils/tower.py

@@ -4,6 +4,7 @@
 # Author: Yuxin Wu <ppwwyyxxc@gmail.com>
 
 import tensorflow as tf
+import re
 
 __all__ = ['get_current_tower_context', 'TowerContext']