首页 Paddle框架 帖子详情
Input X(0)is not initial
收藏
快速回复
Paddle框架 问答深度学习 959 0
Input X(0)is not initial
收藏
快速回复
Paddle框架 问答深度学习 959 0

我用了官方的mnist的例子,然后现在我想输入一个图片,然后写一个预测程序,显示出来每个数字的概率,代码如下,请大神帮看看问题在哪?

import paddle.fluid as fluid
import numpy
import paddle # 导入paddle模块
from PIL import Image

train_pic = paddle.dataset.mnist.train()
pic_data = []
count = 0
for id,data in enumerate(train_pic()):
if count == 1:
break
pic_data = data
count = count+1

print(pic_data[0])

 


img = fluid.layers.data(name='img', shape=[1,28,28], dtype='float32')

predict = fluid.layers.fc(input=img, size=10, act='softmax')

cpu = fluid.core.CPUPlace()
exe = fluid.Executor(cpu)

fluid.io.load_inference_model("recognize_digits.inference.model",
exe,
model_filename=None,
params_filename=None)


prog = fluid.default_main_program()

exe.run(prog)
res=[]
outs = exe.run(
feed={'img': pic_data[0]},
fetch_list=res)

print("----------------------------")
print(res)

0
收藏
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户