Tensorflow 从图像 url 预测？

问题描述

我想根据图片网址进行预测。过去，我使用 ImageDatagenerator().flow_from_directory() 方法，但现在我只有一张图像。所以我想从这张单张图片中进行预测。

我尝试了以下代码，但失败了。 （尺寸错误）

url = "http://3.36.149.28/uploads/WEBUPLOADprofile.png"
img = Image.open(requests.get(url,stream=True).raw)
img = img_to_array(img)
img = img/255.

#Predict
pred = model.predict(img)

所以我尝试重塑和重试，但失败了（无法将大小为 1048576 的数组重塑为形状 (28,28,1)）

img = img.reshape(-1,1)
img = img/255.

#Predict
pred = model.predict(img)

要重塑并获得彩色预测图像，我该怎么办？请帮忙..

附加：我训练了 srcnn 模型和输入：

inputs = Input((None,None,3),dtype='float')

解决方法

我解决了这个问题。

首先，我的 url 图片形状是 (None,None,4)，但我训练的形状是 (None,3)。

所以我尝试了另一个 jpg 图像（无、无、3）并通过 np 扩展尺寸， 和结果形状 = (1,3)

image = np.expand_dims(image,axis=0)
model.predict(image)

来自link

现在我成功获得了预测图像。