image classification web app using stramlit code example

Example: image classification web app using stramlit

import cv2from PIL import Image, ImageOpsimport numpy as npdef import_and_predict(image_data, model):            size = (150,150)            image = ImageOps.fit(image_data, size, Image.ANTIALIAS)        image = np.asarray(image)        img = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)        img_resize = (cv2.resize(img, dsize=(75, 75),    interpolation=cv2.INTER_CUBIC))/255.                img_reshape = img_resize[np.newaxis,...]            prediction = model.predict(img_reshape)                return predictionif file is None:    st.text("Please upload an image file")else:    image = Image.open(file)    st.image(image, use_column_width=True)    prediction = import_and_predict(image, model)        if np.argmax(prediction) == 0:        st.write("It is a paper!")    elif np.argmax(prediction) == 1:        st.write("It is a rock!")    else:        st.write("It is a scissor!")        st.text("Probability (0: Paper, 1: Rock, 2: Scissor")    st.write(prediction)

Tags:

Misc Example