File size: 972 Bytes
d5465e6 b30ce1c d5465e6 b30ce1c d5465e6 b30ce1c d5465e6 b30ce1c d5465e6 b30ce1c d5465e6 cc797bf d5465e6 b30ce1c |
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 |
import cv2
import os
import json
import streamlit as st
@st.cache
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,)
if bgr2rgb:
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
return image
@st.cache
def get_images_list(path_to_folder: str) -> list:
image_names_list = [
x for x in os.listdir(path_to_folder) if x[-3:] in ["jpg", "peg", "png"]
]
return image_names_list
@st.cache
def load_augmentations_config(
path_to_config: str = "configs/augmentations.json",
) -> dict:
with open(path_to_config, "r") as config_file:
augmentations = json.load(config_file)
return augmentations
def get_params_string(param_values: dict) -> str:
params_string = ", ".join(
[k + "=" + str(param_values[k]) for k in param_values.keys()]
)
return params_string
|