File size: 1,705 Bytes
6e7a74d
 
 
 
 
 
b30ce1c
 
 
 
6e7a74d
 
 
 
 
 
 
 
93cade3
6e7a74d
b30ce1c
6e7a74d
 
b30ce1c
a467828
 
 
 
 
 
6e7a74d
 
 
 
e993855
6e7a74d
93cade3
 
 
 
 
 
 
 
 
6e7a74d
 
b30ce1c
e993855
6e7a74d
e993855
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
import streamlit as st

from control import param2func
from utils import get_images_list, load_image


def show_logo():
    st.image(load_image("logo.png", "../images"), format="PNG")


def select_image(path_to_images: str = "images"):
    image_names_list = get_images_list(path_to_images)
    image_name = st.sidebar.selectbox("Select an image:", image_names_list)
    image = load_image(image_name, path_to_images)
    return image


def show_transform_control(transform_params: dict):
    param_values = {"p": 1.0}
    if len(transform_params) == 0:
        st.sidebar.text("Transform has no parameters")
    else:
        for param in transform_params:
            control_function = param2func[param["type"]]
            if isinstance(param["param_name"], list):
                returned_values = control_function(**param)
                for name, value in zip(param["param_name"], returned_values):
                    param_values[name] = value
            else:
                param_values[param["param_name"]] = control_function(**param)
    return param_values


def show_credentials():
    st.markdown("* * *")
    st.subheader("Credentials:")
    st.markdown(
        "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)"
    )
    st.markdown(
        "Image Source: [pexels.com/royalty-free-images](https://pexels.com/royalty-free-images/)"
    )


def show_docstring(obj_with_ds):
    st.markdown("* * *")
    st.subheader("Docstring:")
    st.text(obj_with_ds.__doc__)