Tensorflow 从图像预测 url？

Question

我想根据图像进行预测 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, 28, 28, 1)
img = img/255.

#Predict
pred = model.predict(img)

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

补充：我训练了srcnn模型，输入：

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

Answer 1

我解决了这个问题。

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

所以我尝试了另一个jpg图像（None，None，3）并通过np扩展维度，结果形状 = (1, None, None, 3)

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

来自

现在我成功预测图像了。

Tensorflow 从图像预测 url？

Tensorflow predict from image url?

python

reshape

tensorflow