refactoring, simple transformes added

configs/augmentations.json

@@ -1,9 +1,21 @@
 {
     "ToGray": [],
     "ToSepia": [],
-    "Blur": [{"name" : "blur_limit",
+    "ChannelShuffle": [],
+    "InvertImg": [],
+    "Blur": [{"param_name" : "blur_limit",
               "type" : "int_interval",
-              "limits" : [3, 100],
-              "defaults" : [3, 7]}
+              "limits_list" : [3, 100],
+              "defaults_list" : [3, 7]}
+            ],
+    "CLAHE": [{"param_name" : "clip_limit",
+              "type" : "int_interval",
+              "limits_list" : [1, 100],
+              "defaults_list" : [1, 4]},
+              {"param_name" : "tile_grid_size",
+              "type" : "several_ints",
+              "subparam_names": ["y", "x"],
+              "limits_list" : [[1, 100],[1, 100]],
+              "defaults_list" : [8, 8]}
             ]
 }

src/control.py

@@ -0,0 +1,25 @@
+import streamlit as st
+
+
+def select_int_interval(param_name, limits_list, defaults_list, **kwargs):
+    st.sidebar.subheader(param_name)
+    min_max_interval = st.sidebar.slider('', limits_list[0], limits_list[1], defaults_list)
+    return min_max_interval
+
+def select_several_ints(param_name, subparam_names, limits_list, defaults_list, **kwargs):
+    st.sidebar.subheader(param_name)
+    result = []
+    assert len(limits_list) == len(defaults_list)
+    assert len(subparam_names) == len(defaults_list)
+    
+    for name, limits, defaults in zip(subparam_names, limits_list, defaults_list):
+        result.append(st.sidebar.slider(name, limits[0], limits[1], defaults))
+    return tuple(result)
+
+
+# dict from param name to function showing this param
+param2func = {
+    'int_interval': select_int_interval,
+    'several_ints': select_several_ints,
+    
+}

src/main.py

@@ -8,6 +8,8 @@ import json
 import albumentations as A
 import streamlit as st
 
+from control import *
+
 
 def load_image(image_name, path_to_folder = '../images'):
     path_to_image = os.path.join(path_to_folder, image_name)
@@ -16,19 +18,6 @@ def load_image(image_name, path_to_folder = '../images'):
     return image
 
 
-def show_int_interval(name, limits, defaults, **kwargs):
-    min_max_interval = st.sidebar.slider(name, limits[0], limits[1], defaults)
-    return min_max_interval
-
-
-
-# dict from param name to function showing this param
-param2func = {
-    'int_interval': show_int_interval
-}
-
-
-
 st.title('Demo of Albumentations transforms')
 
 
@@ -55,7 +44,7 @@ else:
         param['value'] = param2func[param['type']](**param)
         
         
-params_string = ','.join([param['name'] + '=' + str(param['value']) for param in transform_params] + ['p=1.0'])
+params_string = ','.join([param['param_name'] + '=' + str(param['value']) for param in transform_params] + ['p=1.0'])
 params_string = '(' + params_string + ')'
 
 st.text('Press R to update')
@@ -63,4 +52,16 @@ exec('transform = A.' + transform_name + params_string)
 st.image([image, transform(image = image)['image']], 
          caption = ['Original image', 'Transformed image'],
          width = 320)
-st.text(str(transform.__doc__))
+
+st.subheader('Docstring:')
+st.text(str(transform.__doc__))
+
+
+st.text('')
+st.text('')
+st.subheader('Credentials:')
+st.text('Source: https://github.com/IliaLarchenko/albumentations-demo')
+st.text('Albumentation library: https://github.com/albumentations-team/albumentations')
+st.text('Image Source: https://www.pexels.com/royalty-free-images/')
+
+