formatting

.gitignore

@@ -57,4 +57,4 @@ docs/_build/
 target/
 
 .ipynb_checkpoints
-\.idea
+\.idea

.pre-commit-config.yaml

@@ -3,6 +3,7 @@ repos:
     rev: v2.3.0
     hooks:
     - id: flake8
+      args: [--max-line-length=88]
     - id: trailing-whitespace
     - id: end-of-file-fixer
     - id: requirements-txt-fixer
@@ -14,4 +15,4 @@ repos:
     rev: 19.3b0
     hooks:
     -   id: black
-        args: [--config=black.toml]
+        args: [--config=black.toml]

Procfile

@@ -1 +1 @@
-web: sh setup.sh && streamlit run src/app.py
+web: sh setup.sh && streamlit run src/app.py

README.md

@@ -24,4 +24,4 @@ And then just follow the address you see in console. You should see something si
 ## Online playground:
 You can also play with the current deployed version:
 [https://albumentations-demo.herokuapp.com/](https://albumentations-demo.herokuapp.com/)
-It is deployed on free service with very limited computing power and can be quite unstable.
+It is deployed on free service with very limited computing power and can be quite unstable.

black.toml

@@ -18,4 +18,4 @@ exclude = '''
     | dist
   )/
 )
-'''
+'''

configs/augmentations.json

@@ -1,503 +1,502 @@
 {
-  "ToGray": [],
-  "ToSepia": [],
-  "ChannelShuffle": [],
-  "InvertImg": [],
-  "VerticalFlip": [],
-  "HorizontalFlip": [],
-  "Flip": [],
-  "Transpose": [],
-  "RandomGridShuffle": [
+  "Blur": [
     {
-      "param_name": "grid",
-      "type": "several_ints",
-      "subparam_names": [
-        "height",
-        "width"
+      "defaults": [
+        3,
+        7
+      ],
+      "limits_list": [
+        3,
+        100
+      ],
+      "param_name": "blur_limit",
+      "type": "int_interval"
+    }
+  ],
+  "CLAHE": [
+    {
+      "defaults": [
+        1,
+        4
+      ],
+      "limits_list": [
+        1,
+        100
+      ],
+      "param_name": "clip_limit",
+      "type": "int_interval"
+    },
+    {
+      "defaults_list": [
+        8,
+        8
       ],
       "limits_list": [
         [
           1,
-          10
+          100
         ],
         [
           1,
-          10
+          100
         ]
       ],
-      "defaults_list": [
-        3,
-        3
-      ]
+      "param_name": "tile_grid_size",
+      "subparam_names": [
+        "height",
+        "width"
+      ],
+      "type": "several_ints"
     }
   ],
-  "Blur": [
+  "CenterCrop": [
     {
-      "param_name": "blur_limit",
-      "type": "int_interval",
+      "param_name": "height",
+      "placeholder": {
+        "defaults": "image_half_height",
+        "limits_list": [
+          1,
+          "image_height"
+        ]
+      },
+      "type": "int_interval"
+    },
+    {
+      "param_name": "width",
+      "placeholder": {
+        "defaults": "image_half_width",
+        "limits_list": [
+          1,
+          "image_width"
+        ]
+      },
+      "type": "int_interval"
+    }
+  ],
+  "ChannelDropout": [
+    {
+      "defaults": [
+        1,
+        1
+      ],
       "limits_list": [
-        3,
-        100
+        1,
+        3
       ],
-      "defaults": [
-        3,
-        7
-      ]
+      "param_name": "channel_drop_range",
+      "type": "int_interval"
+    },
+    {
+      "defaults": 0,
+      "limits_list": [
+        0,
+        255
+      ],
+      "param_name": "fill_value",
+      "type": "int_interval"
     }
   ],
-  "MotionBlur": [
+  "ChannelShuffle": [],
+  "Crop": [
     {
-      "param_name": "blur_limit",
-      "type": "int_interval",
-      "limits_list": [
-        3,
-        100
+      "min_diff": 1,
+      "param_name": [
+        "x_min",
+        "x_max"
       ],
-      "defaults": [
-        3,
-        7
-      ]
+      "placeholder": {
+        "defaults_list": [
+          0,
+          "image_half_width"
+        ],
+        "limits_list": [
+          0,
+          "image_width"
+        ]
+      },
+      "type": "min_max"
+    },
+    {
+      "min_diff": 1,
+      "param_name": [
+        "y_min",
+        "y_max"
+      ],
+      "placeholder": {
+        "defaults_list": [
+          0,
+          "image_half_height"
+        ],
+        "limits_list": [
+          0,
+          "image_height"
+        ]
+      },
+      "type": "min_max"
     }
   ],
-  "MedianBlur": [
+  "Cutout": [
     {
-      "param_name": "blur_limit",
-      "type": "int_interval",
+      "defaults": 8,
       "limits_list": [
-        3,
+        1,
         100
       ],
-      "defaults": [
-        3,
-        7
-      ]
-    }
-  ],
-  "GaussNoise": [
+      "param_name": "num_holes",
+      "type": "int_interval"
+    },
     {
-      "param_name": "var_limit",
-      "type": "int_interval",
+      "defaults": 8,
       "limits_list": [
-        0.0,
-        500.0
+        1,
+        100
       ],
-      "defaults": [
-        10.0,
-        50.0
-      ]
+      "param_name": "max_h_size",
+      "type": "int_interval"
     },
     {
-      "param_name": "mean",
-      "type": "int_interval",
+      "defaults": 8,
       "limits_list": [
-        -100.0,
-        100.0
+        1,
+        100
       ],
-      "defaults": 0.0
+      "param_name": "max_w_size",
+      "type": "int_interval"
+    },
+    {
+      "param_name": "fill_value",
+      "type": "rgb"
     }
   ],
-  "RandomBrightness": [
+  "Equalize": [
     {
-      "param_name": "limit",
-      "type": "int_interval",
-      "limits_list": [
-        -1.0,
-        1.0
+      "options_list": [
+        "cv",
+        "pil"
       ],
-      "defaults": [
-        -0.2,
-        0.2
-      ]
+      "param_name": "mode",
+      "type": "radio"
+    },
+    {
+      "defaults": 1,
+      "param_name": "by_channels",
+      "type": "checkbox"
     }
   ],
-  "RandomContrast": [
+  "Flip": [],
+  "GaussNoise": [
     {
-      "param_name": "limit",
-      "type": "int_interval",
+      "defaults": [
+        10.0,
+        50.0
+      ],
       "limits_list": [
-        -1.0,
-        1.0
+        0.0,
+        500.0
       ],
-      "defaults": [
-        -0.2,
-        0.2
-      ]
+      "param_name": "var_limit",
+      "type": "int_interval"
+    },
+    {
+      "defaults": 0.0,
+      "limits_list": [
+        -100.0,
+        100.0
+      ],
+      "param_name": "mean",
+      "type": "int_interval"
     }
   ],
+  "HorizontalFlip": [],
   "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
       ],
+      "param_name": "hue_shift_limit",
+      "type": "int_interval"
+    },
+    {
       "defaults": [
         -30,
         30
-      ]
-    },
-    {
-      "param_name": "val_shift_limit",
-      "type": "int_interval",
+      ],
       "limits_list": [
         -100,
         100
       ],
+      "param_name": "sat_shift_limit",
+      "type": "int_interval"
+    },
+    {
       "defaults": [
         -20,
         20
-      ]
+      ],
+      "limits_list": [
+        -100,
+        100
+      ],
+      "param_name": "val_shift_limit",
+      "type": "int_interval"
     }
   ],
-  "RGBShift": [
+  "ImageCompression": [
     {
-      "param_name": "r_shift_limit",
-      "type": "int_interval",
-      "limits_list": [
-        -255,
-        255
+      "options_list": [
+        0,
+        1
       ],
-      "defaults": [
-        -20,
-        20
-      ]
+      "param_name": "compression_type",
+      "type": "radio"
     },
     {
-      "param_name": "g_shift_limit",
-      "type": "int_interval",
-      "limits_list": [
-        -255,
-        255
+      "defaults_list": [
+        80,
+        100
       ],
-      "defaults": [
-        -20,
-        20
-      ]
-    },
-    {
-      "param_name": "b_shift_limit",
-      "type": "int_interval",
       "limits_list": [
-        -255,
-        255
+        0,
+        100
       ],
-      "defaults": [
-        -20,
-        20
-      ]
+      "param_name": [
+        "quality_lower",
+        "quality_upper"
+      ],
+      "type": "min_max"
     }
   ],
+  "InvertImg": [],
   "JpegCompression": [
     {
-      "param_name": [
-        "quality_lower",
-        "quality_upper"
+      "defaults_list": [
+        80,
+        100
       ],
-      "type": "min_max",
       "limits_list": [
         0,
         100
       ],
-      "defaults_list": [
-        80,
-        100
-      ]
-    }
-  ],
-  "ImageCompression": [
-    {
-      "param_name": "compression_type",
-      "type": "radio",
-      "options_list": [
-        0,
-        1
-      ]
-    },
-    {
       "param_name": [
         "quality_lower",
         "quality_upper"
       ],
-      "type": "min_max",
+      "type": "min_max"
+    }
+  ],
+  "MedianBlur": [
+    {
+      "defaults": [
+        3,
+        7
+      ],
       "limits_list": [
-        0,
+        3,
         100
       ],
-      "defaults_list": [
-        80,
-        100
-      ]
+      "param_name": "blur_limit",
+      "type": "int_interval"
     }
   ],
-  "CLAHE": [
+  "MotionBlur": [
     {
-      "param_name": "clip_limit",
-      "type": "int_interval",
-      "limits_list": [
-        1,
-        100
-      ],
       "defaults": [
-        1,
-        4
-      ]
-    },
-    {
-      "param_name": "tile_grid_size",
-      "type": "several_ints",
-      "subparam_names": [
-        "height",
-        "width"
+        3,
+        7
       ],
       "limits_list": [
-        [
-          1,
-          100
-        ],
-        [
-          1,
-          100
-        ]
+        3,
+        100
       ],
-      "defaults_list": [
-        8,
-        8
-      ]
+      "param_name": "blur_limit",
+      "type": "int_interval"
     }
   ],
-  "Cutout": [
+  "RGBShift": [
     {
-      "param_name": "num_holes",
-      "type": "int_interval",
+      "defaults": [
+        -20,
+        20
+      ],
       "limits_list": [
-        1,
-        100
+        -255,
+        255
       ],
-      "defaults": 8
+      "param_name": "r_shift_limit",
+      "type": "int_interval"
     },
     {
-      "param_name": "max_h_size",
-      "type": "int_interval",
+      "defaults": [
+        -20,
+        20
+      ],
       "limits_list": [
-        1,
-        100
+        -255,
+        255
       ],
-      "defaults": 8
+      "param_name": "g_shift_limit",
+      "type": "int_interval"
     },
     {
-      "param_name": "max_w_size",
-      "type": "int_interval",
+      "defaults": [
+        -20,
+        20
+      ],
       "limits_list": [
-        1,
-        100
+        -255,
+        255
       ],
-      "defaults": 8
-    },
-    {
-      "param_name": "fill_value",
-      "type": "rgb"
+      "param_name": "b_shift_limit",
+      "type": "int_interval"
     }
   ],
-  "RandomSnow": [
+  "RandomBrightness": [
     {
-      "param_name": [
-        "snow_point_lower",
-        "snow_point_upper"
+      "defaults": [
+        -0.2,
+        0.2
       ],
-      "type": "min_max",
       "limits_list": [
-        0.0,
+        -1.0,
         1.0
       ],
-      "defaults_list": [
-        0.1,
-        0.2
-      ]
-    },
+      "param_name": "limit",
+      "type": "int_interval"
+    }
+  ],
+  "RandomContrast": [
     {
-      "param_name": "brightness_coeff",
-      "type": "int_interval",
+      "defaults": [
+        -0.2,
+        0.2
+      ],
       "limits_list": [
-        0.0,
-        5.0
+        -1.0,
+        1.0
       ],
-      "defaults": 2.5
+      "param_name": "limit",
+      "type": "int_interval"
     }
-
   ],
   "RandomFog": [
     {
-      "param_name": [
-        "fog_coef_lower",
-        "fog_coef_upper"
+      "defaults_list": [
+        0.1,
+        0.2
       ],
-      "type": "min_max",
       "limits_list": [
         0.0,
         1.0
       ],
-      "defaults_list": [
-        0.1,
-        0.2
-      ]
+      "param_name": [
+        "fog_coef_lower",
+        "fog_coef_upper"
+      ],
+      "type": "min_max"
     },
     {
-      "param_name": "alpha_coef",
-      "type": "int_interval",
+      "defaults": 0.08,
       "limits_list": [
         0.0,
         1.0
       ],
-      "defaults": 0.08
+      "param_name": "alpha_coef",
+      "type": "int_interval"
     }
   ],
-  "ChannelDropout": [
+  "RandomGridShuffle": [
     {
-      "param_name": "channel_drop_range",
-      "type": "int_interval",
-      "limits_list": [
-        1,
+      "defaults_list": [
+        3,
         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
-    }
-  ],
-  "CenterCrop": [
-    {
-      "param_name": "height",
-      "type": "int_interval",
-      "placeholder": {
-        "limits_list": [
+        [
           1,
-          "image_height"
+          10
         ],
-        "defaults": "image_half_height"
-      }
-    },
-    {
-      "param_name": "width",
-      "type": "int_interval",
-      "placeholder": {
-        "limits_list": [
+        [
           1,
-          "image_width"
-        ],
-        "defaults": "image_half_width"
-      }
-    }
-  ],
-  "Crop": [
-    {
-      "param_name": [
-        "x_min",
-        "x_max"
-      ],
-      "type": "min_max",
-      "placeholder": {
-        "limits_list": [
-          0,
-          "image_width"
-        ],
-        "defaults_list": [
-          0,
-          "image_half_width"
+          10
         ]
-      },
-      "min_diff": 1
-    },
-    {
-      "param_name": [
-        "y_min",
-        "y_max"
       ],
-      "type": "min_max",
-      "placeholder": {
-        "limits_list": [
-          0,
-          "image_height"
-        ],
-        "defaults_list": [
-          0,
-          "image_half_height"
-        ]
-      },
-      "min_diff": 1
+      "param_name": "grid",
+      "subparam_names": [
+        "height",
+        "width"
+      ],
+      "type": "several_ints"
     }
   ],
   "RandomScale": [
     {
-      "param_name": "scale_limit",
-      "type": "int_interval",
+      "defaults": [
+        -0.1,
+        0.1
+      ],
       "limits_list": [
         -0.9,
         2.0
       ],
-      "defaults": [
-        -0.1,
-        0.1
-      ]
+      "param_name": "scale_limit",
+      "type": "int_interval"
     },
     {
-      "param_name": "interpolation",
-      "type": "radio",
       "options_list": [
         0,
         1,
         2,
         3,
         4
-      ]
+      ],
+      "param_name": "interpolation",
+      "type": "radio"
+    }
+  ],
+  "RandomSnow": [
+    {
+      "defaults_list": [
+        0.1,
+        0.2
+      ],
+      "limits_list": [
+        0.0,
+        1.0
+      ],
+      "param_name": [
+        "snow_point_lower",
+        "snow_point_upper"
+      ],
+      "type": "min_max"
+    },
+    {
+      "defaults": 2.5,
+      "limits_list": [
+        0.0,
+        5.0
+      ],
+      "param_name": "brightness_coeff",
+      "type": "int_interval"
+    }
+  ],
+  "Solarize": [
+    {
+      "defaults": 128,
+      "limits_list": [
+        0,
+        255
+      ],
+      "param_name": "threshold",
+      "type": "int_interval"
     }
-  ]
-}
+  ],
+  "ToGray": [],
+  "ToSepia": [],
+  "Transpose": [],
+  "VerticalFlip": []
+}

run.sh

@@ -1 +1 @@
-streamlit run src/app.py
+streamlit run src/app.py

setup.sh

@@ -10,4 +10,4 @@ echo "\
 headless = true\n\
 enableCORS=false\n\
 port = $PORT\n\
-" > ~/.streamlit/config.toml
+" > ~/.streamlit/config.toml

src/app.py

@@ -48,12 +48,8 @@ augmented_image = transform(image=image)["image"]
 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))

src/utils.py

@@ -10,7 +10,7 @@ def load_image(
     image_name: str, path_to_folder: str = "../images", bgr2rgb: bool = True
 ):
     path_to_image = os.path.join(path_to_folder, image_name)
-    image = cv2.imread(path_to_image,)
+    image = cv2.imread(path_to_image)
     if bgr2rgb:
         image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
     return image

src/visuals.py

@@ -35,13 +35,22 @@ def show_credentials():
     st.markdown("* * *")
     st.subheader("Credentials:")
     st.markdown(
-        "Source: [github.com/IliaLarchenko/albumentations-demo](https://github.com/IliaLarchenko/albumentations-demo)"
+        (
+            "Source: [github.com/IliaLarchenko/albumentations-demo]"
+            "(https://github.com/IliaLarchenko/albumentations-demo)"
+        )
     )
     st.markdown(
-        "Albumentations library: [github.com/albumentations-team/albumentations](https://github.com/albumentations-team/albumentations)"
+        (
+            "Albumentations library: [github.com/albumentations-team/albumentations]"
+            "(https://github.com/albumentations-team/albumentations)"
+        )
     )
     st.markdown(
-        "Image Source: [pexels.com/royalty-free-images](https://pexels.com/royalty-free-images/)"
+        (
+            "Image Source: [pexels.com/royalty-free-images]"
+            "(https://pexels.com/royalty-free-images/)"
+        )
     )