将图像上传使用Streamlit numpy数组

Question

我正在为一个使用Streamlit的机器学习项目开发一个web应用程序。

我在load_img:

遇到了这个错误

TypeError: expected str, bytes or os.PathLike object, not UploadedFile.

from tensorflow.keras.models import load_model
from tensorflow.keras.preprocessing.image import load_img
from tensorflow.keras.preprocessing.image import img_to_array
from tensorflow.keras.applications.vgg16 import preprocess_input
from tensorflow.keras.applications.vgg16 import decode_predictions
from tensorflow.keras.applications.vgg16 import VGG16
from tensorflow.keras.models import model_from_json
import numpy as np
import cv2
import tensorflow as tf
import streamlit as st

st.write("This is a simple image classification web app to identify cars")
file = st.file_uploader("Please upload an image file", type=["jpg", "png"])

def import_and_predict(image_data, model):
image = load_img(image_data,target_size=(64,64))  
img = img_to_array(image)
img = np.array(image) 
img = img / 255.0
image = img.resize((64,64))
img = img.reshape(1,64,64,3)
label = model.predict_classes(img)

prediction = label[0][0]

return f"Prediction: {prediction}" 
if file is None:
st.text("Please upload an image file")
else:
import_and_predict(file, model)

Answer 1

来自file_uploader的Streamlit文档:

>>> uploaded_file = st.file_uploader("Choose a file")
>>> if uploaded_file is not None:
...     # To read file as bytes:
...     bytes_data = uploaded_file.getvalue()
...     st.write(bytes_data)

上面的代码将为您提供一个BytesIO对象，然后可以将其转换为表示图像的数组。