IliaLarchenko's picture
added shape changing transforms support
52a0611
raw
history blame
1.85 kB
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
def fill_placeholders(params, placeholder_params):
# TODO: refactor
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
@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 get_params_string(param_values: dict) -> str:
params_string = ", ".join(
[k + "=" + str(param_values[k]) for k in param_values.keys()]
)
return params_string