Implement square resize

Author: daniel-gallo

main.py

@@ -11,6 +11,7 @@
     RandomCrop,
     RandomHorizontalFlip,
     RandomResizedCrop,
+    SquareResize,
 )
 from transformations.transformation import Transformation
 
@@ -49,6 +50,8 @@ def get_transformation(transformation: str) -> Transformation:
         return RandomCrop(
             scale=(0.4, 1.0), ratio=(0.75, 1.33), min_num_pixels=224 * 224
         )
+    elif transformation == "square_resize":
+        return SquareResize(224)
     else:
         raise NotImplementedError()
 

transformations/__init__.py

@@ -3,11 +3,13 @@
 from .random_crop import RandomCrop
 from .random_horizontal_flip import RandomHorizontalFlip
 from .random_resized_crop import RandomResizedCrop
+from .square_resize import SquareResize
 
 __all__ = [
     AutoAugment,
     NativeAspectRatioResize,
     RandomCrop,
     RandomHorizontalFlip,
     RandomResizedCrop,
+    SquareResize,
 ]

transformations/square_resize.py

@@ -0,0 +1,13 @@
+import tensorflow as tf
+
+from transformations.transformation import Transformation
+
+
+class SquareResize(Transformation):
+    def __init__(self, size: int):
+        self.size = tf.constant([size, size])
+
+    def __call__(self, image):
+        return tf.image.resize(
+            image, self.size, method=tf.image.ResizeMethod.BICUBIC, antialias=True
+        )