File size: 1,984 Bytes
96a96d9
6e7a74d
 
 
 
 
 
b30ce1c
 
 
 
96a96d9
6e7a74d
96a96d9
 
 
 
 
 
 
 
 
6e7a74d
 
dac9a79
93cade3
6e7a74d
b30ce1c
6e7a74d
 
b30ce1c
a467828
 
 
 
 
 
6e7a74d
 
 
 
e993855
6e7a74d
93cade3
13270de
 
 
 
93cade3
 
13270de
 
 
 
93cade3
 
13270de
 
 
 
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
import cv2
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):
    image_names_list = get_images_list(path_to_images)
    if len(image_names_list) < 1:
        return 0, 0
    else:
        try:
            image_name = st.sidebar.selectbox("Select an image:", image_names_list)
            image = load_image(image_name, path_to_images)
            return 1, image
        except cv2.error:
            return 0, 0


def show_transform_control(transform_params: dict) -> 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__)