interior-design / config.py
BertChristiaens's picture
Duplicate from BertChristiaens/controlnet-seg-backup
3d4d894
raw
history blame
1.05 kB
"""File with configs"""
from palette import COLOR_MAPPING_, COLOR_MAPPING
HEIGHT = 512
WIDTH = 512
COLOR_NAMES = list(COLOR_MAPPING.keys())
COLOR_RGB = [tuple(int(k[i:i+2], 16) for i in (1, 3, 5)) for k in COLOR_MAPPING_.keys()] + [(0, 0, 0), (255, 255, 255)]
INVERSE_COLORS = {v: k for k, v in COLOR_MAPPING_.items()}
def map_colors(color: str) -> str:
"""Map color to hex value.
Args:
color (str): color name
Returns:
str: hex value
"""
return COLOR_MAPPING[color]
POS_PROMPT = "tree, sky, cloud, scenery, outdoors, grass, flowers, sunlight, beautiful, ultra detailed beautiful landscape, architectural renderings vegetation, high res, best high quality landscape, outdoor lighting, sunshine, 4k, 8k, realistic"
NEG_PROMPT= "lowres, deformed, blurry, bad anatomy, disfigured, poorly drawn face, mutation, mutated, extra limb, ugly, poorly drawn hands, missing limb, blurry, floating limbs, disconnected limbs, malformed hands, blur, out of focus, long neck, long body, mutated hands and fingers, out of frame"