a lot of new augs

configs/augmentations.json

@@ -3,11 +3,76 @@
     "ToSepia": [],
     "ChannelShuffle": [],
     "InvertImg": [],
+    "VerticalFlip": [],
+    "HorizontalFlip": [],
+    "Flip": [],
+    "RandomGridShuffle": [
+              {"param_name" : "grid",
+              "type" : "several_ints",
+              "subparam_names": ["y", "x"],
+              "limits_list" : [[1, 10],[1, 10]],
+              "defaults_list" : [3, 3]}
+            ],    
     "Blur": [{"param_name" : "blur_limit",
               "type" : "int_interval",
               "limits_list" : [3, 100],
               "defaults" : [3, 7]}
             ],
+    "MotionBlur": [{"param_name" : "blur_limit",
+              "type" : "int_interval",
+              "limits_list" : [3, 100],
+              "defaults" : [3, 7]}
+            ],
+    "MedianBlur": [{"param_name" : "blur_limit",
+              "type" : "int_interval",
+              "limits_list" : [3, 100],
+              "defaults" : [3, 7]}
+            ],
+    "GaussNoise": [{"param_name" : "var_limit",
+              "type" : "int_interval",
+              "limits_list" : [0.0, 500.0],
+              "defaults" : [10.0, 50.0]},
+                   {"param_name" : "mean",
+              "type" : "int_interval",
+              "limits_list" : [-100.0, 100.0],
+              "defaults" : 0.0}
+            ],
+    "RandomBrightness": [{"param_name" : "limit",
+              "type" : "int_interval",
+              "limits_list" : [-1.0, 1.0],
+              "defaults" : [-0.2, 0.2]}
+            ],
+    "RandomContrast": [{"param_name" : "limit",
+              "type" : "int_interval",
+              "limits_list" : [-1.0, 1.0],
+              "defaults" : [-0.2, 0.2]}
+            ],
+    "HueSaturationValue": [{  "param_name" : "hue_shift_limit",
+                              "type" : "int_interval",
+                              "limits_list" : [-100, 100],
+                              "defaults" : [-20, 20]},
+                           {  "param_name" : "sat_shift_limit",
+                              "type" : "int_interval",
+                              "limits_list" : [-100, 100],
+                              "defaults" : [-30, 30]},
+                           {  "param_name" : "val_shift_limit",
+                              "type" : "int_interval",
+                              "limits_list" : [-100, 100],
+                              "defaults" : [-20, 20]}
+                          ],
+    "RGBShift": [{  "param_name" : "r_shift_limit",
+                              "type" : "int_interval",
+                              "limits_list" : [-255, 255],
+                              "defaults" : [-20, 20]},
+                           {  "param_name" : "g_shift_limit",
+                              "type" : "int_interval",
+                              "limits_list" : [-255, 255],
+                              "defaults" : [-20, 20]},
+                           {  "param_name" : "b_shift_limit",
+                              "type" : "int_interval",
+                              "limits_list" : [-255, 255],
+                              "defaults" : [-20, 20]}
+                          ],
     "JpegCompression": [{"param_name" : "quality_lower",
                           "type" : "int_interval",
                           "limits_list" : [0, 100],
@@ -64,8 +129,42 @@
                   "defaults" : 0.1},
                  {"param_name" : "brightness_coeff",
                   "type" : "int_interval",
-                  "limits_list" : [0.9, 20.0],
+                  "limits_list" : [0.0, 20.0],
                   "defaults" : 2.5}
-                ]
-
+                ],
+    "RandomFog": [{"param_name" : "fog_coef_lower",
+                  "type" : "int_interval",
+                  "limits_list" : [0.0, 1.0],
+                  "defaults" : 0.1},
+                   {"param_name" : "fog_coef_upper",
+                  "type" : "int_interval",
+                  "limits_list" : [0.0, 1.0],
+                  "defaults" : 0.1},
+                 {"param_name" : "alpha_coef",
+                  "type" : "int_interval",
+                  "limits_list" : [0.0, 1.0],
+                  "defaults" : 0.08}
+                ],
+    "ChannelDropout": [{"param_name" : "channel_drop_range",
+                  "type" : "int_interval",
+                  "limits_list" : [1, 3],
+                  "defaults" : [1, 1]},
+                  {"param_name" : "fill_value",
+                  "type" : "int_interval",
+                  "limits_list" : [0, 255],
+                  "defaults" : 0}
+                ],
+    "Solarize": [{"param_name" : "threshold",
+              "type" : "int_interval",
+              "limits_list" : [0, 255],
+              "defaults" : 128}
+            ],
+    "Equalize": [
+                 {"param_name" : "mode",
+                  "type" : "radio",
+                  "options_list" : ["cv", "pil"]},
+                 {"param_name" : "by_channels",
+                  "type" : "checkbox",
+                  "defaults": 1}        
+            ]
 }

src/control.py

@@ -25,18 +25,27 @@ def select_several_RGB(param_name, **kwargs):
     return tuple(result)
 
 
-
 def select_radio(param_name, options_list, **kwargs):
     st.sidebar.subheader(param_name)
     result = st.sidebar.radio('', options_list)
+    if isinstance(options_list[0],str):
+        result = '"' + result + '"'
+    return result
+
+
+def select_checkbox(param_name, defaults, **kwargs):
+    st.sidebar.subheader(param_name)
+    result = st.sidebar.checkbox("True", defaults)
     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
+    'rgb': select_several_RGB,
+    'checkbox': select_checkbox
 
 }

src/main.py

@@ -32,7 +32,7 @@ image = load_image(image_name, path_to_images)
 path_to_config = 'configs/augmentations.json'
 with open(path_to_config, 'r') as config_file:
     augmentations = json.load(config_file)
-transform_name = st.sidebar.selectbox('Select a transformation:', list(augmentations.keys()))
+transform_name = st.sidebar.selectbox('Select a transformation:', sorted(list(augmentations.keys())))
 transform_params = augmentations[transform_name]
 
 
@@ -44,9 +44,10 @@ else:
         param['value'] = param2func[param['type']](**param)
         
         
-params_string = ','.join([param['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(transform_name + params_string)
 st.text('Press R to update')
 exec('transform = A.' + transform_name + params_string)
 st.image([image, transform(image = image)['image']],