Onekozmonots/Image_Segmentation_Project

1import streamlit as st2import tensorflow as tf3import tensorflow_datasets as tfds4from tensorflow.keras.models import Model, load_model5from tensorflow.keras.layers import Conv2D, MaxPooling2D, UpSampling2D, Input, concatenate6import numpy as np7import cv28from PIL import Image9import matplotlib.pyplot as plt10 11# Modeli yükleme12@st.cache_resource13def load_unet_model():14    return load_model("unet_model.h5")15 16model = load_unet_model()17 18st.title("MNIST Image Segmentation with U-Net")19 20uploaded_file = st.file_uploader("Bir görüntü yükleyin", type=["jpg", "png", "jpeg"])21 22if uploaded_file is not None:23    image = Image.open(uploaded_file).convert("L")  # Griye çevir24    image = image.resize((32, 32))25    image = np.array(image) / 255.0  # Normalize etme26    image = np.expand_dims(image, axis=[0, -1])  # Model giriş formatına uygun hale getirme27    28    prediction = model.predict(image)29    30    st.image(uploaded_file, caption="Yüklenen Görüntü", use_column_width=True)31    st.image(prediction.squeeze(), caption="Segmentasyon Sonucu", use_column_width=True, clamp=True)32