| import streamlit as st |
| import numpy as np |
| import cv2 |
| from PIL import Image |
| import torch |
| import torchvision.transforms as T |
| from torchvision.models.segmentation import deeplabv3_resnet101 |
| from io import BytesIO |
|
|
| st.set_page_config(page_title="Kleidung umfärben", layout="centered") |
| st.title("🧥 Kleidung umfärben – Color Changer") |
|
|
| uploaded_file = st.file_uploader("📤 Bild hochladen", type=["jpg", "jpeg", "png"]) |
|
|
| @st.cache_resource |
| def load_model(): |
| model = deeplabv3_resnet101(pretrained=True).eval() |
| return model |
|
|
| def segment_clothes(img_pil): |
| transform = T.Compose([ |
|
|
