|
import cv2 |
|
import os |
|
import json |
|
|
|
import streamlit as st |
|
|
|
|
|
@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_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 load_augmentations_config( |
|
placeholder_params: dict, path_to_config: str = "configs/augmentations.json" |
|
) -> dict: |
|
with open(path_to_config, "r") as config_file: |
|
augmentations = json.load(config_file) |
|
for name, params in augmentations.items(): |
|
params = [fill_placeholders(param, placeholder_params) for param in params] |
|
return augmentations |
|
|
|
|
|
def fill_placeholders(params, placeholder_params): |
|
|
|
if "placeholder" in params: |
|
placeholder_dict = params["placeholder"] |
|
for k, v in placeholder_dict.items(): |
|
if isinstance(v, list): |
|
params[k] = [] |
|
for element in v: |
|
if element in placeholder_params: |
|
params[k].append(placeholder_params[element]) |
|
else: |
|
params[k].append(element) |
|
else: |
|
if v in placeholder_params: |
|
params[k] = placeholder_params[v] |
|
else: |
|
params[k] = v |
|
params.pop("placeholder") |
|
return params |
|
|
|
|
|
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 |
|
|
