|
|
|
"""Untitled1.ipynb |
|
|
|
Automatically generated by Colab. |
|
|
|
Original file is located at |
|
https://colab.research.google.com/drive/1AAiRPNd-Nnhg1OZNqQdo0_vdvVyOqala |
|
""" |
|
|
|
import streamlit as st |
|
from tensorflow.keras.models import load_model |
|
from PIL import Image, ImageOps |
|
import numpy as np |
|
|
|
|
|
np.set_printoptions(suppress=True) |
|
|
|
|
|
model = load_model("keras_model.h5", compile=False) |
|
|
|
|
|
class_names = open("labels.txt", "r").readlines() |
|
|
|
def predict_image(image_path): |
|
|
|
|
|
|
|
data = np.ndarray(shape=(1, 224, 224, 3), dtype=np.float32) |
|
|
|
|
|
image = Image.open(image_path).convert("RGB") |
|
|
|
|
|
size = (224, 224) |
|
image = ImageOps.fit(image, size, Image.Resampling.LANCZOS) |
|
|
|
|
|
image_array = np.asarray(image) |
|
|
|
|
|
normalized_image_array = (image_array.astype(np.float32) / 127.5) - 1 |
|
|
|
|
|
data[0] = normalized_image_array |
|
|
|
|
|
prediction = model.predict(data) |
|
index = np.argmax(prediction) |
|
class_name = class_names[index] |
|
confidence_score = prediction[0][index] |
|
|
|
return class_name[2:], confidence_score |
|
|
|
st.title("Image Classification App") |
|
st.write("Upload an image and the app will predict its class.") |
|
|
|
uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"]) |
|
if uploaded_file is not None: |
|
image = Image.open(uploaded_file) |
|
st.image(image, caption='Uploaded Image.', use_column_width=True) |
|
class_name, confidence_score = predict_image(uploaded_file) |
|
st.write("Class:", class_name) |
|
st.write("Confidence Score:", confidence_score) |