refactoring, new transforms

configs/augmentations.json

@@ -6,16 +6,52 @@
     "Blur": [{"param_name" : "blur_limit",
               "type" : "int_interval",
               "limits_list" : [3, 100],
-              "defaults_list" : [3, 7]}
+              "defaults" : [3, 7]}
+            ],
+    "JpegCompression": [{"param_name" : "quality_lower",
+                          "type" : "int_interval",
+                          "limits_list" : [0, 100],
+                          "defaults" : 99},
+                        {"param_name" : "quality_upper",
+                          "type" : "int_interval",
+                          "limits_list" : [0, 100],
+                          "defaults" : 100}
+            ],
+    "ImageCompression": [{"param_name" : "quality_lower",
+                          "type" : "int_interval",
+                          "limits_list" : [0, 100],
+                          "defaults" : 99},
+                        {"param_name" : "quality_upper",
+                          "type" : "int_interval",
+                          "limits_list" : [0, 100],
+                          "defaults" : 100},
+                         {"param_name" : "compression_type",
+                          "type" : "radio",
+                          "options_list" : [0, 1]}
             ],
     "CLAHE": [{"param_name" : "clip_limit",
               "type" : "int_interval",
               "limits_list" : [1, 100],
-              "defaults_list" : [1, 4]},
+              "defaults" : [1, 4]},
               {"param_name" : "tile_grid_size",
               "type" : "several_ints",
               "subparam_names": ["y", "x"],
               "limits_list" : [[1, 100],[1, 100]],
               "defaults_list" : [8, 8]}
+            ],
+    "Cutout": [{"param_name" : "num_holes",
+              "type" : "int_interval",
+              "limits_list" : [1, 100],
+              "defaults" : 8},
+               {"param_name" : "max_h_size",
+              "type" : "int_interval",
+              "limits_list" : [1, 100],
+              "defaults" : 8},
+               {"param_name" : "max_w_size",
+              "type" : "int_interval",
+              "limits_list" : [1, 100],
+              "defaults" : 8},
+              {"param_name" : "fill_value",
+              "type" : "rgb"}
             ]
 }

src/control.py

@@ -1,11 +1,12 @@
 import streamlit as st
 
 
-def select_int_interval(param_name, limits_list, defaults_list, **kwargs):
+def select_int_interval(param_name, limits_list, defaults, **kwargs):
     st.sidebar.subheader(param_name)
-    min_max_interval = st.sidebar.slider('', limits_list[0], limits_list[1], defaults_list)
+    min_max_interval = st.sidebar.slider('', limits_list[0], limits_list[1], defaults, key = hash(param_name))
     return min_max_interval
 
+
 def select_several_ints(param_name, subparam_names, limits_list, defaults_list, **kwargs):
     st.sidebar.subheader(param_name)
     result = []
@@ -13,13 +14,29 @@ def select_several_ints(param_name, subparam_names, limits_list, 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))
+        result.append(st.sidebar.slider(name, limits[0], limits[1], defaults, key = hash(param_name + name)))
+    return tuple(result)
+
+def select_several_RGB(param_name, **kwargs):
+    result = select_several_ints(param_name, 
+                                 subparam_names = ["Red", "Green", "Blue"], 
+                                 limits_list = [[0, 255],[0, 255],[0, 255]],
+                                 defaults_list = [0, 0, 0])
     return tuple(result)
 
 
+
+def select_radio(param_name, options_list, **kwargs):
+    st.sidebar.subheader(param_name)
+    result = st.sidebar.radio('', options_list)
+    return result
+
+
 # dict from param name to function showing this param
 param2func = {
     'int_interval': select_int_interval,
     'several_ints': select_several_ints,
-    
+    'radio': select_radio,
+    'rgb': select_several_RGB
+
 }

src/main.py

@@ -62,6 +62,4 @@ 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/')
-
-
+st.text('Image Source: https://www.pexels.com/royalty-free-images/')