运行多输入模型的时候,报错:Segmentation fault
Created by: BlainWu
自己的模型训练的时候是要输入两个参数,在paddle上训练是正常的。但是当预测的时候,到predictor.run()会报错:“Segmentation fault” 我的定义的预测函数是这样,是不是输入有问题?在文档中没有找到多输入的例子,不知道是预测函数有问题还是模型有问题 `
def predict(predictor, image,vel):
img = image#输入图像数据
img = img.reshape(1, 3, 128, 128)
speed = []
speed.append(vel)
input_img = predictor.get_input(0)
input_speed = predictor.get_input(1)
input_img.resize((1, 3, 128, 128))#设置接口图像大小
input_img.set_data(img)
input_speed.set_data(speed)
predictor.run()
out = predictor.get_output(0)
score = out.data()[0]
print(out.data()[0])
return score`
看到有个类似的问题#3851,也没有回答