title: Albumentations Demo emoji: 🏢 colorFrom: blue colorTo: pink sdk: streamlit sdk_version: 1.22.0 python_version: 3.8 app_file: "src/app.py" pinned: true license: mit short_description: "Optimize image augmentations with Albumentations" tags: ["Computer Vision", "Image Augmentation", "Albumentations", "Streamlit", "Image Processing", "CV", "Image", "Augmentation"]

Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

albumentations-demo

This service is created to demonstrate abilities of the Albumentations - a library for efficient image augmentations.

Link to my article about augmentations selection and why this service can be useful

Project repository: github.com/IliaLarchenko/albumentations-demo

Online demo: huggingface.co/spaces/IliaLarchenko/albumentations-demo

Easy start

If you would like to run the service locally follow the installation instruction.

Installation and run

git clone https://github.com/IliaLarchenko/albumentations-demo
cd albumentations-demo
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 -- --image_folder PATH_TO_YOUR_IMAGE_FOLDER

If your images have some unusual proportions you can use image_width parameter to set the width in pixels of the original image to show. The width of the transformed image and heights of both images will be computed automatically. Default value of width is 400.

streamlit run src/app.py -- --image_width INT_VALUE_OF_WIDTH

In your terminal you will see the link to the running local service similar to :

  You can now view your Streamlit app in your browser.

  Network URL: http://YOUR_LOCAL_IP:8501
  External URL: http://YOUR_GLOBAL_IP:8501

Just follow the local link to use the service.

Run in docker

You can run the service in docker:

docker-compose up

It will be available at http://DOCKER_HOST_IP:8501

How to use

The interface is very simple and intuitive: 1. On the left you have a control sidebar. Select the "Simple" mode. You can choose the image and the transformation. 2. After that you will see the control elements for the every parameter this transformation has. 3. Every time you change any parameter you will see the augmented version of the image on the right side of your screen. 4. Below the images you can find a code for calling of the augmentation with selected parameters. 5. You can also find there the original docstring for this transformation.

Professional mode

In the professional mode you can: 1. Upload your own image 2. Combine multiple transformations 3. See the random parameters used to get the result

Be aware that in Professional mode some combination of parameters of different transformations can be invalid. You should control it.

Links

• Albumentations library: github.com/albumentations-team/albumentations
• Image Source: pexels.com/royalty-free-images
• Streamlit - framework powering this app github.com/streamlit/streamlit