added the optional path to the images folder

README.md

@@ -16,6 +16,11 @@ pip install -r requirements.txt
 streamlit run src/app.py
 ```
 
+If you want to work with you own images just replace the last line with:
+```
+streamlit run src/app.py -- --images_folder PATH_TO_YOUR_IMAGE_FOLDER
+```
+
 In your terminal you will see the link to the running local service similar to : 
 ```console
   You can now view your Streamlit app in your browser.
@@ -39,4 +44,4 @@ The interface is very simple and intuitive:
 ## Links
 * Albumentations library: [github.com/albumentations-team/albumentations](https://github.com/albumentations-team/albumentations)
 * Image Source: [pexels.com/royalty-free-images](https://pexels.com/royalty-free-images/)
-* Streamlit - framework powering this app [github.com/streamlit/streamlit](https://github.com/streamlit/streamlit)  
+* Streamlit - framework powering this app [github.com/streamlit/streamlit](https://github.com/streamlit/streamlit)  

configs/augmentations.json

@@ -507,6 +507,58 @@
       "type": "checkbox"
     }
   ],
+  "OpticalDistortion": [
+    {
+      "defaults": [
+        -0.3,
+        0.3
+      ],
+      "limits_list": [
+        -2.0,
+        2.0
+      ],
+      "param_name": "distort_limit",
+      "type": "num_interval"
+    },
+    {
+      "defaults": [
+        -0.05,
+        0.05
+      ],
+      "limits_list": [
+        -1.0,
+        1.0
+      ],
+      "param_name": "shift_limit",
+      "type": "num_interval"
+    },
+    {
+      "options_list": [
+        0,
+        1,
+        2,
+        3,
+        4
+      ],
+      "param_name": "interpolation",
+      "type": "radio"
+    },
+    {
+      "options_list": [
+        0,
+        1,
+        2,
+        3,
+        4
+      ],
+      "param_name": "border_mode",
+      "type": "radio"
+    },
+    {
+      "param_name": "value",
+      "type": "rgb"
+    }
+  ],
   "Posterize": [
     {
       "defaults_list": [
@@ -1109,54 +1161,5 @@
   "ToGray": [],
   "ToSepia": [],
   "Transpose": [],
-  "VerticalFlip": [],
-
-
-  "OpticalDistortion": [
-    {
-      "defaults": [
-        -0.3,
-        0.3
-      ],
-      "limits_list": [
-        -2.0,
-        2.0
-      ],
-      "param_name": "distort_limit",
-      "type": "num_interval"
-    },    {
-      "defaults": [-0.05, 0.05],
-      "limits_list": [
-        -1.0,
-        1.0
-      ],
-      "param_name": "shift_limit",
-      "type": "num_interval"
-    },
-    {
-      "options_list": [
-        0,
-        1,
-        2,
-        3,
-        4
-      ],
-      "param_name": "interpolation",
-      "type": "radio"
-    },
-    {
-      "options_list": [
-        0,
-        1,
-        2,
-        3,
-        4
-      ],
-      "param_name": "border_mode",
-      "type": "radio"
-    },
-    {
-      "param_name": "value",
-      "type": "rgb"
-    }]
+  "VerticalFlip": []
 }

src/app.py

@@ -1,7 +1,9 @@
+import os
 import streamlit as st
 import albumentations as A
 
-from utils import load_augmentations_config
+
+from utils import load_augmentations_config, get_path_to_the_image
 from visuals import (
     show_transform_control,
     select_image,
@@ -9,53 +11,62 @@ from visuals import (
     show_docstring,
 )
 
+# get the path to images
+path_to_images = get_path_to_the_image()
+if not os.path.isdir(path_to_images):
+    st.title("There is no directory: " + path_to_images)
+else:
+    status, image = select_image(path_to_images)
+    if status == 0:
+        st.title("Can't load image from: " + path_to_images)
+    else:
+        # show title
+        st.title("Demo of Albumentations")
 
-# show title
-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),
-}
+        # select image
 
-# load the config
-augmentations = load_augmentations_config(
-    placeholder_params, "configs/augmentations.json"
-)
+        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),
+        }
 
-# select a transformation
-transform_name = st.sidebar.selectbox(
-    "Select a transformation:", sorted(list(augmentations.keys()))
-)
+        # load the config
+        augmentations = load_augmentations_config(
+            placeholder_params, "configs/augmentations.json"
+        )
 
-# select the params values
-param_values = show_transform_control(augmentations[transform_name])
+        # select a transformation
+        transform_name = st.sidebar.selectbox(
+            "Select a transformation:", sorted(list(augmentations.keys()))
+        )
 
-# apply the transformation to the image
-transform = getattr(A, transform_name)(**param_values)
-data = A.ReplayCompose([transform])(image=image)
-augmented_image = data["image"]
+        # select the params values
+        param_values = show_transform_control(augmentations[transform_name])
 
-# TODO add convinient replay compose
-# applied_params = data["replay"]["transforms"][0]['params']
-# for k,v in applied_params.items():
-#     applied_params[k] = str(v)
-# st.write(applied_params)
-# st.write(data["replay"])
+        # apply the transformation to the image
+        transform = getattr(A, transform_name)(**param_values)
+        data = A.ReplayCompose([transform])(image=image)
+        augmented_image = data["image"]
 
+        # TODO add convinient replay compose
+        # applied_params = data["replay"]["transforms"][0]['params']
+        # for k,v in applied_params.items():
+        #     applied_params[k] = str(v)
+        # st.write(applied_params)
+        # st.write(data["replay"])
 
-# show the images
-width_original = 400
-width_transformed = int(width_original / image.shape[1] * augmented_image.shape[1])
+        # 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(augmented_image, caption="Transformed image", width=width_transformed)
+        st.image(image, caption="Original image", width=width_original)
+        st.image(augmented_image, caption="Transformed image", width=width_transformed)
 
-# print additional info
-st.code(str(transform))
-show_docstring(transform)
-show_credentials()
+        # print additional info
+        st.code(str(transform))
+        show_docstring(transform)
+        show_credentials()

src/utils.py

@@ -1,10 +1,19 @@
 import cv2
 import os
 import json
+import argparse
 
 import streamlit as st
 
 
+@st.cache
+def get_path_to_the_image():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--images_folder", default="images")
+    args = parser.parse_args()
+    return getattr(args, "images_folder")
+
+
 @st.cache
 def get_images_list(path_to_folder: str) -> list:
     """Return the list of images from folder
@@ -18,9 +27,7 @@ def get_images_list(path_to_folder: str) -> list:
 
 
 @st.cache
-def load_image(
-    image_name: str, path_to_folder: str = "../images", bgr2rgb: bool = True
-):
+def load_image(image_name: str, path_to_folder: str, bgr2rgb: bool = True):
     """Load the image
     Args:
         image_name (str): name of the image

src/visuals.py

@@ -1,3 +1,4 @@
+import cv2
 import streamlit as st
 
 from control import param2func
@@ -8,11 +9,17 @@ def show_logo():
     st.image(load_image("logo.png", "../images"), format="PNG")
 
 
-def select_image(path_to_images: str = "images"):
+def select_image(path_to_images: str):
     image_names_list = get_images_list(path_to_images)
-    image_name = st.sidebar.selectbox("Select an image:", image_names_list)
-    image = load_image(image_name, path_to_images)
-    return image
+    if len(image_names_list) < 1:
+        return 0, 0
+    else:
+        try:
+            image_name = st.sidebar.selectbox("Select an image:", image_names_list)
+            image = load_image(image_name, path_to_images)
+            return 1, image
+        except cv2.error:
+            return 0, 0
 
 
 def show_transform_control(transform_params: dict) -> dict: