如何解决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
现在我成功获得了预测图像。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。