[dataset] added a transform to resize images

bob/learn/pytorch/datasets/__init__.py

@@ -6,6 +6,7 @@ from .utils import FaceCropper
 from .utils import RollChannels 
 from .utils import ToTensor 
 from .utils import Normalize 
+from .utils import Resize 
 
 from .utils import map_labels
 from .utils import ConcatDataset

bob/learn/pytorch/datasets/utils.py

@@ -54,6 +54,20 @@ class Normalize(object):
     sample['image'] = self.op(sample['image'])
     return sample
 
+class Resize(object):
+  def __init__(self, size):
+    self.op = transforms.Resize(size)
+
+  def __call__(self, sample):
+    # convert to PIL image
+    from PIL.Image import fromarray
+    img = fromarray(sample['image'].squeeze())
+    img = self.op(img)
+    sample['image'] = numpy.array(img)
+    sample['image'] = sample['image'][..., numpy.newaxis]
+    return sample
+
+
 def map_labels(raw_labels, start_index=0):
   """
   Map the ID label to [0 - # of IDs]