added shape changing transforms support

configs/augmentations.json

@@ -6,6 +6,7 @@
   "VerticalFlip": [],
   "HorizontalFlip": [],
   "Flip": [],
+  "Transpose": [],
   "RandomGridShuffle": [
     {
       "param_name": "grid",
@@ -421,6 +422,24 @@
       "param_name": "by_channels",
       "type": "checkbox",
       "defaults": 1
+    }],
+    "CenterCrop" :
+    [
+      {
+        "param_name": "height",
+        "type": "int_interval",
+        "placeholder": {
+          "limits_list" : [1, "image_height"],
+          "defaults" : "image_half_height"
+        }
+      },
+      {
+        "param_name": "width",
+        "type": "int_interval",
+        "placeholder": {
+          "limits_list" : [1, "image_width"],
+          "defaults" : "image_half_width"
+        }
+      }
+    ]
     }
-  ]
-}

src/app.py

@@ -1,10 +1,7 @@
 import streamlit as st
 import albumentations as A
 
-from utils import (
-    load_augmentations_config,
-    get_params_string
-)
+from utils import load_augmentations_config
 from visuals import (
     show_transform_control,
     select_image,
@@ -14,19 +11,28 @@ from visuals import (
 
 
 # show title
-st.title("Demo of Albumentations transforms")
+st.title("Demo of Albumentations")
 
 # select image
 image = select_image(path_to_images="images")
+placeholder_params = {
+    "image_width": image.shape[1],
+    "image_height": image.shape[0],
+    "image_half_width": int(image.shape[1] / 2),
+    "image_half_height": int(image.shape[0] / 2),
+}
 
 # load the config
-augmentations = load_augmentations_config("configs/augmentations.json")
+augmentations = load_augmentations_config(
+    placeholder_params, "configs/augmentations.json"
+)
 
 # select a transformation
 transform_name = st.sidebar.selectbox(
     "Select a transformation:", sorted(list(augmentations.keys()))
 )
 
+
 # select the params values
 param_values = show_transform_control(augmentations[transform_name])
 
@@ -40,10 +46,14 @@ augmented_image = transform(image=image)["image"]
 # st.text("Press R to update")
 
 # show the images
+width_original = 400
+width_transformed = int(width_original / image.shape[1] * augmented_image.shape[1])
+
+st.image(
+    image, caption="Original image", width=width_original,
+)
 st.image(
-    [image, augmented_image],
-    caption=["Original image", "Transformed image"],
-    width=335,
+    augmented_image, caption="Transformed image", width=width_transformed,
 )
 
 # print additional info

src/control.py

@@ -26,7 +26,7 @@ def select_several_ints(
     return tuple(result)
 
 
-def select_several_RGB(param_name, **kwargs):
+def select_RGB(param_name, **kwargs):
     result = select_several_ints(
         param_name,
         subparam_names=["Red", "Green", "Blue"],
@@ -53,6 +53,6 @@ param2func = {
     "int_interval": select_int_interval,
     "several_ints": select_several_ints,
     "radio": select_radio,
-    "rgb": select_several_RGB,
+    "rgb": select_RGB,
     "checkbox": select_checkbox,
 }

src/utils.py

@@ -24,12 +24,35 @@ def get_images_list(path_to_folder: str) -> list:
     return image_names_list
 
 
+def fill_placeholders(params, placeholder_params):
+    # TODO: refactor
+    if "placeholder" in params:
+        placeholder_dict = params["placeholder"]
+        for k, v in placeholder_dict.items():
+            if isinstance(v, list):
+                params[k] = []
+                for element in v:
+                    if element in placeholder_params:
+                        params[k].append(placeholder_params[element])
+                    else:
+                        params[k].append(element)
+            else:
+                if v in placeholder_params:
+                    params[k] = placeholder_params[v]
+                else:
+                    params[k] = v
+        params.pop("placeholder")
+    return params
+
+
 @st.cache
 def load_augmentations_config(
-    path_to_config: str = "configs/augmentations.json",
+    placeholder_params: dict, path_to_config: str = "configs/augmentations.json"
 ) -> dict:
     with open(path_to_config, "r") as config_file:
         augmentations = json.load(config_file)
+    for name, params in augmentations.items():
+        params = [fill_placeholders(param, placeholder_params) for param in params]
     return augmentations