我试图在 django 中执行 model.predict(image),但出现错误
我在这里编码
import keras.api.models
from keras.api.preprocessing.image import load_img, img_to_array
from keras.api.applications.vgg16 import VGG16, preprocess_input
import numpy as np
def image_predict(image_data):
image = image_data
image = load_img(image, target_size=(224, 224))
image = img_to_array(image)
image = np.array(image)
image = preprocess_input(image)
image = np.reshape(image, (1, 224, 224, 3))
prediction = np.argmax(nermodel.predict(image)[0])
return prediction
我获取一个图像并将其转换为np数组,我不知道如果通过load_img()获取数据出现问题是否会导致错误
我已经下载了代码并检查了图像如何变化,据我所知,图像加载正确
我循环使用图像路径,对吗?
#example
images = Image.objects.filter(user_id=user)
for image in images:
image_list.append(image.image.path)
for img in image_list:
result = image_predict(img)
预测应该等于例如 3
运行 model.predict(image) 行时出现错误
我已经尝试更改编码、快速 API 和更改数据集。
型号是
>>> model
<Sequential name=sequential, built=True>
我找到了答案,这个模型在cmd中返回进度条,这是django无法编码或解码的,问题的解决方案:
model.predict(image, verbose=0)