image-matting / utils /functions.py
szk1ck's picture
update
62f41f9
raw
history blame
1.7 kB
import os, random
from logging import getLogger, StreamHandler, DEBUG
logger = getLogger(__name__)
handler = StreamHandler(); handler.setLevel(DEBUG)
logger.setLevel(DEBUG)
logger.addHandler(handler)
logger.propagate = False
def get_random_name():
famous_painters = [
"Leonardo", "DaVinci",
"Michelangelo",
"Pablo", "Picasso",
"Vincent", "VanGogh",
"Rembrandt", "VanRijn",
"Claude", "Monet",
"Salvador", "Dali",
"Jackson", "Pollock",
"Andy", "Warhol",
"Henri", "Matisse",
"Georgia", "Keeffe",
"Edvard", "Munch",
"Wassily", "Kandinsky",
"Gustav", "Klimt",
"Rene", "Magritte",
"Frida", "Kahlo",
"Edgar", "Degas",
"Johannes", "Vermeer",
"Paul", "Cezanne",
"Marc", "Chagall",
]
random_painter = random.choice(famous_painters)
# 4桁の乱数を生成
rand_num = random.randint(1000, 9999)
return random_painter+str(rand_num)
def complete(work_dir):
work_dir = work_dir
# logger.debug(f"complete :", work_dir)
return work_dir
def clean(text_output):
# logger.debug(f"text_output : {text_output}")
if text_output!="idle_state":
logger.info(f"clean up : {text_output}.zip")
os.remove(f"{text_output}.zip")
return "idle_state"
else:
logger.info(f"reset")
return "idle_state"
def clean_by_name(text_output):
# logger.debug(f"text_output : {text_output}")
if text_output!="idle_state":
text_output, dir_name = text_output.split("+")
logger.info(f"clean up : {dir_name}.zip")
os.remove(f"{dir_name}.zip")
return "idle_state"
else:
logger.info(f"reset")
return "idle_state"