张量流从图片网址预测？答案

【问题标题】：Tensorflow predict from image url?张量流从图片网址预测？
【发布时间】：2021-03-28 03:53:57
【问题描述】：

我想从图片网址进行预测。过去，我使用 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')

【问题讨论】：

火车图像的大小是多少？或者，model.summary() 对输入的形状有何看法？

标签： python tensorflow reshape

【解决方案1】：

我解决了这个问题。

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

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

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

来自link

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

【讨论】：